Minor fixes - heroku deployment