Menu highlighting now works