improving plotting
improving plotting