Add more stuff for heroku