Added state update