That makes sense. What would be the best way to make it possible to run it with `%pylab inline` or `%matplotlib inline` that would work in 3.0?