Update for heroku deploy