Gradient descent test case, notebook 2, code cell 4