added the models and model relationships