Minor update, fixes for heroku