Minor changes for heroku