2 files changed, 14 insertions, 3 deletions

diff --git a/js/document/views/base.js b/js/document/views/base.js
index db72cc60..d13dce1a 100755
--- a/js/document/views/base.js
+++ b/js/document/views/base.js
@@ -28,7 +28,7 @@ exports.BaseDocumentView = Montage.create(Component, {
         value: null
     },
     ////////////////////////////////////////////////////////////////////
-        //
+        //TODO: This should be renamed to better illustrate it's a container (iframe for design, div for code view)
         iframe: {
                 get: function() {return this._iframe;},
         set: function(value) {this._iframe= value;}
diff --git a/js/document/views/design.js b/js/document/views/design.js
index 13f89692..d772aa86 100755
--- a/js/document/views/design.js
+++ b/js/document/views/design.js
@@ -113,7 +113,7 @@ exports.DesignDocumentView = Montage.create(BaseDocumentView, {
         render: {
         value: function (callback, template) {
                 //TODO: Remove, this is a temp patch for webRequest API gate
-                this.application.ninja.documentController._hackRootFlag = false;
+                this.application.ninja.documentController.redirectRequests = false;
                 //Storing callback for dispatch ready
                 this._callback = callback;
                 this._template = template;
@@ -131,7 +131,8 @@ exports.DesignDocumentView = Montage.create(BaseDocumentView, {
         //
     onTemplateLoad: {
         value: function (e) {
-                this.application.ninja.documentController._hackRootFlag = true;
+                //console.log(this.iframe.contentWindow);
+                this.application.ninja.documentController.redirectRequests = true;
                 //TODO: Add support to constructing URL with a base HREF
                 var basetag = this.content.document.getElementsByTagName('base');
                 //Removing event
@@ -279,6 +280,16 @@ exports.DesignDocumentView = Montage.create(BaseDocumentView, {
             } else {
                 //Else there is not data to parse
             }
+            //TODO: Verify appropiate location for this operation
+                if (this._template && this._template.type === 'banner') {
+                        this.model.documentRoot = this.document.body.getElementsByTagName('ninja-content')[0];
+                } else {
+                        this.model.documentRoot = this.document.body;
+                }
+                //Storing node list for reference (might need to store in the model)
+                this._liveNodeList = this.model.documentRoot.getElementsByTagName('*');
+                //Initiliazing document model
+                document.application.njUtils.makeElementModel(this.model.documentRoot, "Body", "body");
                 //Makign callback if specified
                 if (this._callback) this._callback();
         }