[Kwant] Implementing the BHZ model