Added relationships in the models