1 files changed, 15 insertions, 2 deletions

diff --git a/imports/codemirror/mode/stex/stex.js b/imports/codemirror/mode/stex/stex.js
index bb47fb45..b89e619e 100644
--- a/imports/codemirror/mode/stex/stex.js
+++ b/imports/codemirror/mode/stex/stex.js
@@ -82,7 +82,7 @@ CodeMirror.defineMode("stex", function(cmCfg, modeCfg)
     }
 
     function normal(source, state) {
-        if (source.match(/^\\[a-z]+/)) {
+        if (source.match(/^\\[a-zA-Z@]+/)) {
             var cmdName = source.current();
             cmdName = cmdName.substr(1, cmdName.length-1);
             var plug = plugins[cmdName];
@@ -95,9 +95,22 @@ CodeMirror.defineMode("stex", function(cmCfg, modeCfg)
             return plug.style;
         }
 
+        // escape characters 
+        if (source.match(/^\\[$&%#{}_]/)) {
+          return "tag";
+        }
+
+        // white space control characters
+        if (source.match(/^\\[,;!\/]/)) {
+          return "tag";
+        }
+
         var ch = source.next();
         if (ch == "%") {
-            setState(state, inCComment);
+            // special case: % at end of its own line; stay in same state
+            if (!source.eol()) {
+              setState(state, inCComment);
+            }
             return "comment";
         } 
         else if (ch=='}' || ch==']') {