Make it work with Python3