added models and relationships