Tree Map UI enhancement