Fix ipympl example

[martinRenou]

For some reason the 3D example did not work properly, going to a simpler example works better.