Compatible with python 3
Merge pull request #4 from SteelyWing/patch-1

Compatible with python 3