Make the code compatibale with both, python2 and python3
Merge pull request #8 from thiblahute/python3

Make the code compatible with both, python2 and python3