added a few things to make it work in heroku