Changed back models.py to clean the pull request