Added KL divergence class