Update test description and fix pep8.