Update TreeGenerator.class