Vendor gem sorting

Would it be possible to add a red / blue / green gem button on the vendors side to highlight gems specific to a socket? I can't seem to make a regex to do it.

It could work similar to the gem tab but just highlighting instead of removing other colours would work as well.

Just a QOL thought.