fix a bunch of math bugs in the jensen-shannon estimator