Updates needed for heroku