Updates for heroku deployment