Remove pylab import