Add TODO to models.py