Make sure to use np.array