changed the models.py