Add spec for state model