Add token class diagram