Make changes for heroku