Add schemas and relationships to models