cleaned imports on TreeBuilder