Refactored the decision tree, lemma stress with reduce