Is there a way to color user defined keywords?

By default the keyword “TODO” has it’s own color in Atom, among various other programming keywords.

I find this feature very handy for writing todo lists, although, it doesn’t extend to other keywords such as “WIP” and “DONE” (as shown in image). So I was wondering if there’s a way to expand it to custom keywords not recognized by the IDE?