Formatted and pythonized

[mzechmeister]

Finally, a pure python version. Thanks! Here is a proposal to format the code and to use some more pythonic features.

[shbhuk]

Thank you @mzechmeister . Included these changes