reformatting of models.py