Add uml state machine for menus