Update package for python3+ compatibility