We're trying to tidy up the existing examples to avoid pylab where possible. In this case, would you mind taking a look at re-factoring this example to use ```matplotlib.pyplot as plt```?