Kl Divergence

Implement different KL versions