Correctly (I think) generates the graph now! Check it oooooout