Add the state map stuff