Added changes to deploy to Heroku