Refactored name of save_graph to save_2d_graph