Added support for python3