Made modification in hopes of working with Heroku