remove unused imports
pep8 fixes