Add offset info for partial sort also