Gradient descent optimization is added