Fix decision tree implementation