Correction to plot waves over the range 0 to 2pi