changes to classifier, kl div