Updates to KLDivergence, made by Nathan.