Add update function to heroku