ADD weight calculations
ADD CountTree draft, needs improvement for high dimensional datasets.