Fixed plot_wavefield() for few input parameters