Added support to Python 3