Add method to set the token, fix the documentation