Removed pylab import