[Kwant] Issue in display wave function in 3D lattice