Fixed bug in product kernel gradients diag wrt to X