Convert the script to Python 3

[anuveyatsu]

Currently is written for Python 2 but if we want to automate the run via GH Actions we need Python 3: https://github.blog/changelog/2020-02-27-github-actions-breaking-change-python-2-being-removed-from-all-virtual-environments/