Made pep8 related changes to script.