changed numpy import