Update sanity_model.py