[Kwant] Drawing edge states in zGNR (using tutorial 2.8)