Add support for highlighting 'key' nodes.

ATM this is used to show the main occurrence of math environments.