Add some stuff for heroku