improve pep8