treemap example cleanup