Right now, we rely on the state machine interface to matplotlib. We should instead have the simulation class create and store a figure object and then work from that. Then dovis() can be made to return the figure object if we want to interact with it in a notebook.