Add KL divergence