Add python3 support