Minor changes to models.py