Update plot_prof.py