add to the correct token method