Add tokenization module