Add zero_extend() function