Small changes for heroku deploy