Added relationships to controller