Added wrap_key_part interface