Fix python 3 issue
Merge pull request #3 from joharei/master

Fix python 3 issue