Updating IState to extend IEnter and IExit