Fixed models.py so it can actually be used