Decision trees in Scala