Set initial wavefunction to 2D gaussian