Update Models.py

Updated models according to schema