Better UI Experience for Replacing Tokens

Right now, you can get into the context menu of a token and there is only a UI difference for the selected token when you are hovering over it. You use the background change when you are inside the context menu. When you’re doing something like replacing the token with another token you have, it takes a couple of seconds to find the token you might want to replace it with. This is long enough so you might forget which token you originally selected. When you go back to look at the list of token values in the Left Panel, it isn’t clear which token you just replaced since the item no longer has any UI changes. The impact is that you might try to change a token you’ve already changed since you forgot which one you were one if you are changing a lot at once.
I think a fix could keep the selection on the item until the context closes; or there could be a notch on the context menu that lines up with the selected token.