Add basic functionality to enable menu highlighting