Add test case for special tokens