2 files changed, 13 insertions, 1 deletions

diff --git a/js/document/html-document.js b/js/document/html-document.js
index 8f9d2870..42a7d537 100755
--- a/js/document/html-document.js
+++ b/js/document/html-document.js
@@ -340,6 +340,12 @@ exports.HTMLDocument = Montage.create(TextDocument, {
                                                 shapeModel.slope = root._slope;
                                                 break;
 
+                    case root.GEOM_TYPE_BRUSH_STROKE:
+                        elementModel.selection = "BrushStroke";
+                        elementModel.pi = "BrushStrokePi";
+                                                break;
+
+
                                         default:
                                                 console.log( "geometry type not supported for file I/O, " + root.geomType());
                                                 break;
diff --git a/js/document/text-document.js b/js/document/text-document.js
index 88464d87..54a53999 100755
--- a/js/document/text-document.js
+++ b/js/document/text-document.js
@@ -163,7 +163,13 @@ var TextDocument = exports.TextDocument = Montage.create(Component, {
 
     needsSave: {
         get: function() { return this._needsSave; },
-        set: function(value) { this._needsSave = value }
+        set: function(value) {
+            var i=0, canSaveAll=false;
+            this._needsSave = value;
+
+            this.application.ninja.documentController.canSave = value;//for save menu state update
+            this.application.ninja.documentController.canSaveAll = value;//for save menu state update..currently save all is tried to the current document only
+        }
     },
 
     callback: {