fixing models.py