improve plots for lens_charge