Changed list to np.array for better manipulation