Changed code to heroku's example