Updated the models.py file and associated files.