solve_bellman_equation modified so that sup norm is used to check the convergence.