removed unused imports and pep8 reformat