messily added the submenu highlight feature