Update notes for submitted changes

* Use the `state_graph.py` for all states diagrams