Thanks, I think it's a great mod too. It's shocking to me that this type of thing doesn't exist by default, so I felt very compelled to create it.
Your idea is a fine one and if I was better at making mods I am sure that could be accomplished. However, I have no experience making mods and really just put this together because it HAD to exist. I shared it because I knew how good it was.
Anyone with a lot of experience in mod making is free to send me the code change and I will incorporate your suggestion.
Thanks!