Add states for thinking about location