Added mlp covariance (x gradient not working)