Update for push to heroku