updated the models.py file