I am attempting to plot an animated density profile for the 1D Nonlinear Schrödinger equation https://en.wikipedia.org/wiki/Nonlinear_Schr%C3%B6dinger_equation with generali