Added relationships to models

Added relationships to the models