Figured out how to do the tokenization without slightly modifying the features