diff --git a/octoprint_enclosure/static/js/enclosure.js b/octoprint_enclosure/static/js/enclosure.js
index 3c247aa..fc883be 100644
--- a/octoprint_enclosure/static/js/enclosure.js
+++ b/octoprint_enclosure/static/js/enclosure.js
@@ -137,7 +137,7 @@ $(function () {
 
     });
 
-    self.fromInputData = function (data) {
+    self.fromOutputData = function (data) {
 
       self.isNew(data === undefined);
 
diff --git a/octoprint_enclosure/templates/enclosure_settings.jinja2 b/octoprint_enclosure/templates/enclosure_settings.jinja2
index f76a0bd..c4abac7 100644
--- a/octoprint_enclosure/templates/enclosure_settings.jinja2
+++ b/octoprint_enclosure/templates/enclosure_settings.jinja2
@@ -11,4 +11,8 @@
 {% include "input_editor/main_screen.jinja2" %}
 
 
-<button class="btn pull-right" data-bind="click: function() { $root.print_data(); }">Test</button>
\ No newline at end of file
+
+
+<!-- <div class="control-group">
+    <button class="btn pull-right" data-bind="click: function() { $root.print_data(); }">Test</button>
+</div> -->
\ No newline at end of file
diff --git a/octoprint_enclosure/templates/input_editor/main_screen.jinja2 b/octoprint_enclosure/templates/input_editor/main_screen.jinja2
index b89e373..04f30e1 100644
--- a/octoprint_enclosure/templates/input_editor/main_screen.jinja2
+++ b/octoprint_enclosure/templates/input_editor/main_screen.jinja2
@@ -21,33 +21,32 @@
 
                 <li><a href="#settings_inputs_edit_dialog_temperature_sensor"
                         data-toggle="tab">{{ _('Temperature Sensor') }}</a></li>
-
-
             </ul>
 
             <div class="tab-content">
                 <div id="settings_inputs_edit_dialog_general" class="tab-pane active">
-                    <!-- {% include "input_folder/general_info.jinja2" %} -->
+                    {% include "input_editor/general_info.jinja2" %}
                 </div>
 
                 <div id="settings_inputs_edit_dialog_io" class="tab-pane">
-                    <!-- {% include "input_folder/io_output.jinja2" %} -->
+                    {% include "input_editor/io_output.jinja2" %}
                 </div>
 
                 <div id="settings_inputs_edit_dialog_printer_action" class="tab-pane">
-                    <!-- {% include "input_folder/printer_action.jinja2" %} -->
+                    {% include "input_editor/printer_action.jinja2" %}
                 </div>
 
                 <div id="settings_inputs_edit_dialog_output_action" class="tab-pane">
-                    <!-- {% include "input_folder/output_action.jinja2" %} -->
+                    {% include "input_editor/output_action.jinja2" %}
                 </div>
 
-                 <!-- ko if: ($data.input_type() == "temperature_sensor")  -->
+                <!-- ko if: (inputEditor.input_type() == "temperature_sensor")  -->
                 <div id="settings_inputs_edit_dialog_temperature_sensor" class="tab-pane">
-                    <!-- {% include "input_folder/temperature_sensor.jinja2" %} -->
+                    {% include "input_editor/temperature_sensor.jinja2" %}
                 </div>
                 <!-- /ko -->
 
+
             </div>
         </div>