Make changes for heroku deploy