small change to save function for heroku