Remove unused pylab import