Decision Tree class in Scala