revising based on Jon's comments
updated models.py