Add python3 support in example