Cleanup with pyflakes