Trying with different grids. The value function doesn't converge  with (relatively) large values of sigma_b. Need to check the update_rule