Update gen_json_graph.ipynb