improved tree UI