Add comments in graph.go