Update the code in models.py.