add setToken function