Gradient task 3 fixed