Adding a PlotAmplitude module to plot the amplitudes