Merge pull request #2 from peacedove/master

Update Sankey function