Better style for models.py