edit models.py to get tests running #6