added changes to run on heroku