minor edits on models.py