Revert "Update models.py"

This reverts commit 7b541076ce39d6a6e11db64a5df45adc7c516a2f.
Revert "Revert "Update models.py""

This reverts commit e614e9235ede380080f266e15401f771ef55c282.