Minor change models.py