Use same keyword detection as Atom's

Well, I'm not sure it's exactly the same but it gets closer to it.

It basically allows syntaxes like `// TODO ...` while it ignores others like `// UNCHANGED: ...`