pep8 fixes
fixing imports