models.py now fully reflects our example