bug fix

python2 python3 support