Lets say after 10 iterations or diving, the robot catched the ball. You need to stop the iterator. I guess you add a new feature and test for it in that for loop?

Also this is a toy example - can you show example of real code? handling many states?