Adding a graph generation step