minor edit in models.py