Updated the models to include required relations.