Add Python 2 compatibility
Merge pull request #1 from thomasballinger/python2-compat

Add Python 2 compatibility