added some plotting stuff