[Kwant] Saving a figure of the system