Added support for optional weight correction in BallKMeans.