Sankey: clean-up and add comment