Improved section highlighting