Add heroku compatibility