added plot of wave packet