Cleanup gradient_descent