tidying up models.py