Updated the file to work with heroku