diff --git a/octoprint_CalibrationTools/EStepsApi.py b/octoprint_CalibrationTools/EStepsApi.py
index a9b559f..d8ae183 100644
--- a/octoprint_CalibrationTools/EStepsApi.py
+++ b/octoprint_CalibrationTools/EStepsApi.py
@@ -1,11 +1,17 @@
+# -*- coding: utf-8 -*-
+from __future__ import absolute_import, division, print_function, unicode_literals
+
+import re
 from threading import Event
 
+import flask
+
 CMD_ESP_TEST = "TEST"
 CMD_ESP_LOAD_STEPS = "eSteps_load"
 CMD_ESP_START_EXTRUSION = "eSteps_startExtrusion"
 CMD_ESP_SAVE = "eSteps_save"
 
-class API(octoprint.plugin.SimpleApiPlugin):
+class API:
 
     @staticmethod
     def apiCommands():
diff --git a/octoprint_CalibrationTools/__init__.py b/octoprint_CalibrationTools/__init__.py
index 33dfd7e..c11c13f 100644
--- a/octoprint_CalibrationTools/__init__.py
+++ b/octoprint_CalibrationTools/__init__.py
@@ -43,10 +43,9 @@ class CalibrationtoolsPlugin(
         # Define your plugin's asset files to automatically include in the
         # core UI here.
         return {
-            "js": ["js/CalibrationTools.js"],
+            "js": ["js/CalibrationTools.js","js/PIDTuneViewModel.js"],
             "css": ["css/style.css"]
         }
-    # self._settings.valid_boolean_trues:
 
     def get_settings_defaults(self):
         return defaultSettings
diff --git a/octoprint_CalibrationTools/api.py b/octoprint_CalibrationTools/api.py
index 46c2004..7d0dffd 100644
--- a/octoprint_CalibrationTools/api.py
+++ b/octoprint_CalibrationTools/api.py
@@ -7,15 +7,21 @@ from threading import Event
 import flask
 import octoprint.plugin
 
+# from octoprint_CalibrationTools import EStepsApi
+
 CMD_TEST = "TEST"
 CMD_LOAD_STEPS = "loadSteps"
 CMD_START_EXTRUSION = "startExtrusion"
 CMD_SAVE_E_STEPS = "saveESteps"
 
-
 class API(octoprint.plugin.SimpleApiPlugin):
+    def __init__(self) -> None:
+        super().__init__()
+        # self.espAPI = EStepsApi.API()
+
     @staticmethod
     def get_api_commands():
+        # x = EStepsApi.apiCommands()
         return {
             CMD_LOAD_STEPS: [],
             CMD_START_EXTRUSION: [],
@@ -33,6 +39,8 @@ class API(octoprint.plugin.SimpleApiPlugin):
 
     def on_api_command(self, command, data):
         self._logger.debug("api command [%s] received payload [%s]", command, data)
+        self.espAPI.apiGateWay()
+
         if command == CMD_LOAD_STEPS:
             self._logger.debug("Load steps from EEPROM")
             if not self._printer.is_ready():
diff --git a/octoprint_CalibrationTools/static/js/CalibrationTools.js b/octoprint_CalibrationTools/static/js/CalibrationTools.js
index c14bbbd..569824a 100644
--- a/octoprint_CalibrationTools/static/js/CalibrationTools.js
+++ b/octoprint_CalibrationTools/static/js/CalibrationTools.js
@@ -41,13 +41,6 @@ $(function () {
             self.is_admin(self.loginStateViewModel.isAdmin());
         }
 
-        /**open settings*/
-        self.openCalibrationSettings = function () {
-            $('a#navbar_show_settings').click();
-            $('li#settings_plugin_CalibrationTools_link a').click();
-            $("#settings_plugin_CalibrationTools").click();
-        }
-
         self.from_json = function (response) {
             self.steps["X"](response.data.X);
             self.steps["Y"](response.data.Y);
@@ -83,10 +76,6 @@ $(function () {
             })
         }
 
-        self.tempRestart = function () {
-            OctoPrint.system.executeCommand("core", "restart");
-        }
-
         self.saveESteps = function () {
             OctoPrint.simpleApiCommand("CalibrationTools", "saveESteps", {
                 "newESteps": self.results.newSteps()
@@ -105,19 +94,6 @@ $(function () {
                 self.from_json(response);
             });
         }
-
-        self.test = function () {
-            OctoPrint.simpleApiCommand("CalibrationTools", "TEST").done(function (response) {
-                console.log(response)
-            }).fail(function (response) {
-                new PNotify({
-                    title: "Test request",
-                    text: response.responseJSON.error.msg,
-                    type: "error"
-                });
-            })
-        }
-
     }
 
     // This is how our plugin registers itself with the application, by adding some configuration
diff --git a/octoprint_CalibrationTools/templates/CalibrationTools_tab.jinja2 b/octoprint_CalibrationTools/templates/CalibrationTools_tab.jinja2
index fdfcacd..a655d1a 100644
--- a/octoprint_CalibrationTools/templates/CalibrationTools_tab.jinja2
+++ b/octoprint_CalibrationTools/templates/CalibrationTools_tab.jinja2
@@ -1,5 +1,5 @@
 <div class="octo-tab-content">
-    <div class="row-fluid header">
+    <div class="row-fluid header" id="toolBar">
         <div class="span3 title">
             <p>Calibration</p>
         </div>
@@ -7,7 +7,7 @@
             <div class="row-fluid">
                 <div class="span6"></div>
                 <div class="span6">
-                    <button class="btn" data-bind="click: $root.tempRestart, visible: true">
+                    <button class="btn" data-bind="click: $root.restart, visible: true">
                         RESTART
                     </button>
                     <button class="btn btn-primary" data-bind="click: $root.test,  visible: true" title="BTN for testing">
@@ -18,7 +18,7 @@
             </div>
         </div>
         <div class="span1 icon">
-            <a data-bind="click: $root.openCalibrationSettings">
+            <a data-bind="click: $root.onSettingsClick">
                 <i class="fas fa-cog fa-lg" data-color="#ddd"></i>
             </a>
         </div>
@@ -44,4 +44,34 @@
             {% include "tabs/tab-content.jinja2" %}
         </div>
     </div>
-</div>
\ No newline at end of file
+</div>
+
+{% block script%}
+<script type="text/javascript">
+    OCTOPRINT_VIEWMODELS.push({
+        construct: function () {
+            this.restart = function () { OctoPrint.system.executeCommand('core', 'restart') };
+
+            this.test = function () {
+                OctoPrint.simpleApiCommand("CalibrationTools", "TEST").done(function (response) {
+                    console.log(response)
+                }).fail(function (response) {
+                    new PNotify({
+                        title: "Test request",
+                        text: response.responseJSON.error.msg,
+                        type: "error"
+                    });
+                });
+            };
+
+            this.onSettingsClick = function () {
+                $('a#navbar_show_settings').click();
+                $('li#settings_plugin_CalibrationTools_link a').click();
+                $("#settings_plugin_CalibrationTools").click();
+            };
+        },
+        // dependencies: ["terminalViewModel"],
+        elements: [".toolBar"]
+    });
+</script>
+{% endblock %}
\ No newline at end of file