updates to graph generation