Update make_plot.py