Update for heroku deployment