newStates is simpler than before