Updating doc for gradients