Update DecisionTreeRunner.scala