Update HW 6

Adding problem 3 gradient