improve pep8 script