Cleaned models.py file