Add token class