[Kwant] Plotting current density in Kwant