Update TreeBuilder.java