Similar to the colorization PR: would it be possible to add the entropy of the token to the colorization as well for comparison? https://nikkin.dev/blog/llm-entropy.html