Treemap clean up