[FEATURE] add state to model