UI improvement for tree.