models.py edited