Added decision tree data collecting, not tested.