Changed things to run on heroku