Cleaning up with pyflakes