refactored script per pep8 and 2to3