Add numpy import