Add token handler