Add token type functions