Refactor this Python code to make it cleaner and more idiomatic. Avoid changing its behavior.