Pep8 tests. TODO fix