Added relationships to models and database migration