Cosmetic changes to the python script
Merge pull request #38 from wpoa/script-work

Cosmetic changes to the python script