Wiggled code around to pass pep8

Thanks for the script!