Update models.py

Python 3