modifications for decision tree