Revert "Update models.py"

This reverts commit 3781a2a1fb3bf8eade5426fe1bb8ac90c8921940.