Update TreeVisitor.java