Add get token function