Update the KL Divergence Calculation.