Update models.py

Minor edit