Improve the JS script to highlight the current ToC entry.