Added some mods to Heroku for working