adding updated exercise 5.1. I just updated the comment in function numerical_solution