more changes for tree map