pep8: pep8 fix, and remove unused import