Now works in Python3