Revert "Reverted models.py"

This reverts commit 4b4d14aa89c91b6c8574c7259b604de57a028dd0.