corrected models.py