Works with Python3
Finally works on Python2 and 3