improving plotting stuff