2 files changed, 12 insertions, 4 deletions

diff --git a/js/document/models/base.js b/js/document/models/base.js
index c44123c3..a73b8b20 100755
--- a/js/document/models/base.js
+++ b/js/document/models/base.js
@@ -236,6 +236,8 @@ exports.BaseDocumentModel = Montage.create(Component, {
     //
     save: {
         value: function (callback, libCopyCallback) {
+            var self = this;
+
             //TODO: Implement on demand logic
             if (this.needsSave) {
                 //Save
@@ -270,6 +272,15 @@ exports.BaseDocumentModel = Montage.create(Component, {
                 }
             } else if (this.currentView === this.views.code) {
                 //TODO: Add save logic for code view
+                //save to textarea
+                self.views.code.editor.save();
+                //save to disk
+                this.application.ninja.ioMediator.fileSave({
+                    mode: 'html-text',
+                    file: self.file,
+                    content:self.views.code.textArea.value
+                }, this.handleSaved.bind({callback: callback, model: this}));
+
             } else {
                 //TODO: Error handle
             }
diff --git a/js/document/views/design-code.js b/js/document/views/design-code.js
index 44d12549..8ad6d555 100644
--- a/js/document/views/design-code.js
+++ b/js/document/views/design-code.js
@@ -70,10 +70,7 @@ exports.DesignCodeView = Montage.create(CodeDocumentView, {
     //
     show: {
         value: function (callback) {
-
-            this.textViewContainer.style.display = "block";
-            this.textViewContainer.style.background = "white";
-            this.textViewContainer.style.height = "100%";
+            this.textViewContainer.setAttribute("class", "codeViewContainer cm-s-default");
 
             //todo : update options bar