models.py - forgot to commit this along with the previous commit