Modifed models.py