modified models.py