Removed unnecessary pylab import.