fixed models.py