2 files changed, 15 insertions, 16 deletions

diff --git a/js/document/text-document.js b/js/document/text-document.js
index 6f8efaad..f2b7b0d8 100755
--- a/js/document/text-document.js
+++ b/js/document/text-document.js
@@ -191,5 +191,14 @@ var TextDocument = exports.TextDocument = Montage.create(Montage, {
         value: function() {
             // Have the XHR here?
         }
+    }/*
+,
+
+    save:{
+        value:function(){
+            //base function - to be overridden
+        }
     }
+*/
+
 });
\ No newline at end of file
diff --git a/js/mediators/io-mediator.js b/js/mediators/io-mediator.js
index 2008e40f..599b7d84 100644
--- a/js/mediators/io-mediator.js
+++ b/js/mediators/io-mediator.js
@@ -142,7 +142,12 @@ exports.IoMediator = Montage.create(Component, {
                 //
                 switch (file.mode) {
                         case 'html':
-                                contents = this.parseNinjaTemplateToHtml(file);
+                                file.document.content.document.body.innerHTML = file.body;
+                                file.document.content.document.head.innerHTML = file.head;
+                                if (file.style) {
+                                        file.document.content.document.head.getElementsByTagName('style')[0].innerHTML = this.getCssFromRules(file.style.cssRules);
+                                }
+                                contents = file.document.content.document.documentElement.outerHTML;
                                 break;
                         default:
                                 contents = file.content;
@@ -184,21 +189,6 @@ exports.IoMediator = Montage.create(Component, {
         }
     },
     ////////////////////////////////////////////////////////////////////
-    //TODO: Expand to allow more templates
-    parseNinjaTemplateToHtml: {
-        enumerable: false,
-        value: function (template) {
-                //
-                template.document.content.document.body.innerHTML = template.body;
-                template.document.content.document.head.innerHTML = template.head;
-                //TODO: Remove temp fix for styles
-                if (template.style) {
-                        template.document.content.document.head.getElementsByTagName('style')[0].innerHTML = this.getCssFromRules(template.style.cssRules);
-                }
-                return template.document.content.document.documentElement.outerHTML;
-        }
-    },
-    ////////////////////////////////////////////////////////////////////
     //Method to return a string from CSS rules (to be saved to a file)
     getCssFromRules: {
         enumerable: false,