Added plot_waveform