Add support to Python 3