[Kwant] Kwant-Plot discretize potential with different colours