Added relationships to the models