made edits to models.py