Edit script to conform to pep8.