Modified the graph_viewer.py to plot 2 dimensions (x-y)