Convert to work with python 3