Updated my models.py file