Added code to app runs in Heroku