removed uneeded imports
pep8 stuff