Adding ability to plot waves generated by main_sim.py