Fixed things up for heroku