Adding changes for heroku