make plotting working