Updated gradient descent test