Major update for the sankey diagram.