treemap/sankey cleanup and comments