KL divergence version 0