Rename graph_or_task to graph