States module to define states that set contexts