make highlighting work on load, too