feat: Add possibility to override the token class