Added comments to the models.py file