Remove unused imports
Pyflakes cleanup