updated models.py