Small changes for heroku