add python2 and python3 compatibility

currently, python2 is better supported