updated plotting stuff