made some tree UI fixes