highlightAll now accepts an element to query

If no element is passed, the function defaults to querying document, hopefully requiring minimal changes to existing code.