Prepared for Heroku
added models