Added the set_token method