Add tokenize function