Updated code for heroku deployment