Add more functionality to `token_mgr`