Added models and generated the schema