Added support to python 3