Adding full support for unique keys

I'd like a way to either color code (automatically), or annotate after (like right now we have PK and FK etc...can we get something like UK for unique key) the fields that are part of a unique key? Thanks!