Adding some more stuff for Heroku