Support both python2 and python3
Merge pull request #3 from bwrsandman/patch-1

Support both python2 and python3