Added relationships between models