State Machine Pattern Unity