Add support for heroku