Add python 3 support