Make some changes for Heroku