Update create_plots.py