Add unit testing and test tokenizer