Update models2.py