How can I modify the state machine logic? - EdgeBench