Add support for py2.7