import numpy for np.array