And you could just do the np.array inline here...