Small fix to pyplot

Now pyplot redraws itself. It's not how it's going to be implemented in
the final version.