Update TreeInterrogation.scala