Add token type enum