diff --git a/syntax/confile.vim b/syntax/confile.vim
index e5790a5..655f607 100644
--- a/syntax/confile.vim
+++ b/syntax/confile.vim
@@ -1,16 +1,24 @@
 " Vim Syntax file for CON files
 
-syn match SetLength /^\d\{3\}/ nextgroup=SetKeyword,SetKeywordMarked
-syn match SetKeyword /\d\{4\}/ contained
-syn match SetKeywordMarked /310[123]\|5000/ contained nextgroup=SetKeywordMarkedInfo
-syn match SetKeywordMarkedInfo /..*/ contained
-
+syn match SetLength /^\d\{3\}/ nextgroup=SetKeyword,SetKeywordMarked,SetKeywordInfoMarked
 hi SetLength guifg=grey
+
+syn match SetKeyword /\d\{4\}/ contained
 hi SetKeyword guifg=lightgreen
+
+" Visitations marked completely
+syn match SetKeywordMarked /310[123]\|5000/ contained nextgroup=SetKeywordMarkedInfo
 hi SetKeywordMarked gui=inverse guifg=lightgreen
+
+" Treatments only mark treatment info
+syn match SetKeywordInfoMarked /5001/ contained nextgroup=SetKeywordMarkedInfo
+hi SetKeywordInfoMarked guifg=lightgreen
+
+" mark keyword value
+syn match SetKeywordMarkedInfo /..*/ contained
 hi SetKeywordMarkedInfo gui=inverse
 
-
+" Mark datasets like patient
 syn match SetStart /^0138000/ nextgroup=SetType
 syn match SetType /con0\|con9\|besa\|adt0\|adt9\|010[1-4]\|rvsa/ contained " 0101