Nice simplification! It makes more sense to encapsulate this logic in the `run()` function.