Minor changes to make heroku app run