[Kwant] Figsize when ax is enabled in kwant.plot