Get rid of the numpy import