1 files changed, 2 insertions, 1 deletions
diff --git a/kgramstats.h b/kgramstats.h
index 248b193..b40e1ab 100644
--- a/kgramstats.h
+++ b/kgramstats.h

@@ -23,9 +23,10 @@ private:
                int titlecase;
                int uppercase;
                int period;
+                string* token;
        } token_data;
        int maxK;
-        map<kgram, map<string, token_data*>* >* stats;
+        map<kgram, map<int, token_data*>* >* stats;
 };
 void printKgram(kgram k);