Extended edit_graph to include sets