Update 1-D_model_problem.py