add KL-divergence