Small refactoring to make the code more readable.