edited models.py