Add word segmentation module