Modified the code so that it also works in Python 2.7.x
Merge pull request #1 from briandamaged/master

Modified the code so that it also works in Python 2.7.x