Fix vectorization notebook to include %pylab magic