Add python3 compatible version