fix PEP8 and unused imports