[Kwant] Wavefunction for 2D NISIN