changes to plot_graphs.py