Apply flake8 cleanup to puz.py to shorten line lengths