Add python 2 support