While a UML is important to have in your architecture, you can create other simpler graphs that reflect the actions developing in the game. What does the player do? What happens when an enemy dies? What happens when the player dies?
I have seen people use Figma for these types of designs. Perhaps it can be interesting to you.