added plotting stuff