changes for heroku

To make it work with heroku