py2 support
Merge pull request #2 from hayd/py2

py2 support