Add TODO and friends highlighting