Update diagram_graph.rb