Is this different from `arange(n_states)` in any way?