Make highlighting TODOs into a function