got gradient descent to work!!