change the models.py