Remove pylab from import statements