Actually store the states in the map