fix pep8
remove unused import