Cosmetic changes to models.py