Fixed faulty highlighting in TODO and EDGE_CASE