Cleared up the API a bit and added a 2D example for gradient descent.