add relationships in models.py