Create plot_wave.py