Update generate_simple_model.py