Fixed up bregman ball class and wrote tests, fixed KL divergence so it works for unnormalized distributions