update tests
pep8 fixes