nit: python-style wants two spaces before the comment (so `else:  # ...`).