Fix pep8 warnings

issue #1