[Kwant] Graphene dispersion example script