Comment:

A diagram of the state machine would be helpful here. I know it is in another part of the module, but since a reader might only come to this section, it would help if it is repeated here

Summary:

Move state machine diagram