Add python2 support. closes #1