Made changes for heroku