not needed. Would np.array(states) work fine without an if statement.