Remove unused imports
Pep8 the tests package