Update models.py

Added some comments