Update model_checks.py