pep8 fixes, removed unused imports.