remove unused imports + pep8