I thought I saw something like this before, but I don't remember where.
It's good to be able to where the lemma is used in a given context. Repetition in a text is a valuable interpretive tool.
BUT if the lemma is a noun, it would be helpful also to note the corresponding verb, and vice versa, if a verb, the corresponding noun.
e.g. In 2 Corinthians 1, we have the verb παρακαλέω 4 times. We also have the noun παράκλησις 6 times. I would like the option that we can see corresponding roots at the same time too, so all these show up at the same time.