Updated state machine docs after first review round