cleaned up models.py with some comments and a short TODO