added partial support to state update