Update simple_plot.py