fix pep8 Fixes #2