Updated the 'StateMachine' class and its associated tests to integrate the 'State' and 'Transition' classes.