setup.py: PEP 8 fixes
setup.py: Remove unused imports