pep8 fix and update unit test