add kl divergence