Add some JS to automatically make the appropriate toolbar highlight active