Removed conversion to np.array