Add support for py2 and py3.