Update run_the_model.py