[Kwant] Distribution of spin in graphene lattice

Dear Kwant developers and users, For the plot of the graphene wavefunction as illustrated in http://kwant-project.org/doc/1.0/tutorial/tutorial6, is it possible to show the distribution of the spin in the same plot? Should I include the spin orbital coupling? Thanks. Best, Johnny