modified models.py file