Updated the comments to plot_wavefield