added function to get amb_kf cov