do pep8, remove unused import