Add state machine diagrams