Changed end parameter of gen_graph