STYLE: removed unused imports and variables (with pyflakes)