Added models and database