fixing some plotting stuff