Updates to the 2-state model