Fixed Decision Tree code