updated the models and provided the relationships