Update controller and views for heroku