add function for get token