diff --git a/.idea/misc.xml b/.idea/misc.xml
new file mode 100644
index 0000000..65531ca
--- /dev/null
+++ b/.idea/misc.xml
@@ -0,0 +1,4 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="ProjectRootManager" version="2" project-jdk-name="Python 3.6" project-jdk-type="Python SDK" />
+</project>
\ No newline at end of file
diff --git a/.idea/modules.xml b/.idea/modules.xml
new file mode 100644
index 0000000..859de4f
--- /dev/null
+++ b/.idea/modules.xml
@@ -0,0 +1,8 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="ProjectModuleManager">
+    <modules>
+      <module fileurl="file://$PROJECT_DIR$/.idea/new-pysplice.iml" filepath="$PROJECT_DIR$/.idea/new-pysplice.iml" />
+    </modules>
+  </component>
+</project>
\ No newline at end of file
diff --git a/.idea/new-pysplice.iml b/.idea/new-pysplice.iml
new file mode 100644
index 0000000..e98082a
--- /dev/null
+++ b/.idea/new-pysplice.iml
@@ -0,0 +1,12 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<module type="PYTHON_MODULE" version="4">
+  <component name="NewModuleRootManager">
+    <content url="file://$MODULE_DIR$" />
+    <orderEntry type="inheritedJdk" />
+    <orderEntry type="sourceFolder" forTests="false" />
+  </component>
+  <component name="TestRunnerService">
+    <option name="projectConfiguration" value="py.test" />
+    <option name="PROJECT_TEST_RUNNER" value="py.test" />
+  </component>
+</module>
\ No newline at end of file
diff --git a/.idea/vcs.xml b/.idea/vcs.xml
new file mode 100644
index 0000000..94a25f7
--- /dev/null
+++ b/.idea/vcs.xml
@@ -0,0 +1,6 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="VcsDirectoryMappings">
+    <mapping directory="$PROJECT_DIR$" vcs="Git" />
+  </component>
+</project>
\ No newline at end of file
diff --git a/.idea/workspace.xml b/.idea/workspace.xml
new file mode 100644
index 0000000..8ce4114
--- /dev/null
+++ b/.idea/workspace.xml
@@ -0,0 +1,128 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="ChangeListManager">
+    <list default="true" id="c59d642a-84af-4ba9-8ba3-1f6dd760951a" name="Default" comment="">
+      <change beforePath="$PROJECT_DIR$/setup.py" beforeDir="false" afterPath="$PROJECT_DIR$/setup.py" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/splicemachine/ml/zeppelin.py" beforeDir="false" afterPath="$PROJECT_DIR$/splicemachine/ml/zeppelin.py" afterDir="false" />
+    </list>
+    <option name="EXCLUDED_CONVERTED_TO_IGNORED" value="true" />
+    <option name="TRACKING_ENABLED" value="true" />
+    <option name="SHOW_DIALOG" value="false" />
+    <option name="HIGHLIGHT_CONFLICTS" value="true" />
+    <option name="HIGHLIGHT_NON_ACTIVE_CHANGELIST" value="false" />
+    <option name="LAST_RESOLUTION" value="IGNORE" />
+  </component>
+  <component name="FileEditorManager">
+    <leaf />
+  </component>
+  <component name="Git.Settings">
+    <option name="RECENT_GIT_ROOT_PATH" value="$PROJECT_DIR$" />
+  </component>
+  <component name="IdeDocumentHistory">
+    <option name="CHANGED_PATHS">
+      <list>
+        <option value="$PROJECT_DIR$/splicemachine/ml/zeppelin.py" />
+      </list>
+    </option>
+  </component>
+  <component name="ProjectFrameBounds">
+    <option name="y" value="22" />
+    <option name="width" value="1440" />
+    <option name="height" value="795" />
+  </component>
+  <component name="ProjectView">
+    <navigator proportions="" version="1">
+      <foldersAlwaysOnTop value="true" />
+    </navigator>
+    <panes>
+      <pane id="ProjectPane">
+        <subPane>
+          <expand>
+            <path>
+              <item name="new-pysplice" type="b2602c69:ProjectViewProjectNode" />
+              <item name="new-pysplice" type="462c0819:PsiDirectoryNode" />
+            </path>
+            <path>
+              <item name="new-pysplice" type="b2602c69:ProjectViewProjectNode" />
+              <item name="new-pysplice" type="462c0819:PsiDirectoryNode" />
+              <item name="splicemachine" type="462c0819:PsiDirectoryNode" />
+            </path>
+            <path>
+              <item name="new-pysplice" type="b2602c69:ProjectViewProjectNode" />
+              <item name="new-pysplice" type="462c0819:PsiDirectoryNode" />
+              <item name="splicemachine" type="462c0819:PsiDirectoryNode" />
+              <item name="ml" type="462c0819:PsiDirectoryNode" />
+            </path>
+          </expand>
+          <select />
+        </subPane>
+      </pane>
+      <pane id="Scope" />
+    </panes>
+  </component>
+  <component name="PropertiesComponent">
+    <property name="last_opened_file_path" value="$PROJECT_DIR$" />
+  </component>
+  <component name="RunDashboard">
+    <option name="ruleStates">
+      <list>
+        <RuleState>
+          <option name="name" value="ConfigurationTypeDashboardGroupingRule" />
+        </RuleState>
+        <RuleState>
+          <option name="name" value="StatusDashboardGroupingRule" />
+        </RuleState>
+      </list>
+    </option>
+  </component>
+  <component name="SvnConfiguration">
+    <configuration />
+  </component>
+  <component name="TaskManager">
+    <task active="true" id="Default" summary="Default task">
+      <changelist id="c59d642a-84af-4ba9-8ba3-1f6dd760951a" name="Default" comment="" />
+      <created>1533844782895</created>
+      <option name="number" value="Default" />
+      <option name="presentableId" value="Default" />
+      <updated>1533844782895</updated>
+    </task>
+    <servers />
+  </component>
+  <component name="ToolWindowManager">
+    <frame x="0" y="22" width="1440" height="795" extended-state="0" />
+    <editor active="true" />
+    <layout>
+      <window_info active="true" content_ui="combo" id="Project" order="0" visible="true" weight="0.3404864" />
+      <window_info anchor="bottom" id="TODO" order="6" />
+      <window_info anchor="bottom" id="Docker" order="7" show_stripe_button="false" />
+      <window_info anchor="bottom" id="Event Log" order="7" side_tool="true" />
+      <window_info anchor="bottom" id="Version Control" order="7" />
+      <window_info anchor="bottom" id="Python Console" order="7" />
+      <window_info anchor="bottom" id="Run" order="2" />
+      <window_info id="Structure" order="1" side_tool="true" weight="0.25" />
+      <window_info anchor="right" id="Kite" order="3" />
+      <window_info anchor="bottom" id="Terminal" order="7" />
+      <window_info id="Favorites" order="2" side_tool="true" />
+      <window_info anchor="bottom" id="Debug" order="3" weight="0.4" />
+      <window_info anchor="right" content_ui="combo" id="Hierarchy" order="2" weight="0.25" />
+      <window_info anchor="bottom" id="Inspection" order="5" weight="0.4" />
+      <window_info anchor="right" id="Commander" internal_type="SLIDING" order="0" type="SLIDING" weight="0.4" />
+      <window_info anchor="right" id="Ant Build" order="1" weight="0.25" />
+      <window_info anchor="bottom" id="Message" order="0" />
+      <window_info anchor="bottom" id="Cvs" order="4" weight="0.25" />
+      <window_info anchor="bottom" id="Find" order="1" />
+    </layout>
+  </component>
+  <component name="VcsContentAnnotationSettings">
+    <option name="myLimit" value="2678400000" />
+  </component>
+  <component name="editorHistoryManager">
+    <entry file="file://$PROJECT_DIR$/splicemachine/ml/zeppelin.py">
+      <provider selected="true" editor-type-id="text-editor">
+        <state relative-caret-position="623">
+          <caret line="95" column="39" lean-forward="true" selection-start-line="95" selection-start-column="39" selection-end-line="95" selection-end-column="39" />
+        </state>
+      </provider>
+    </entry>
+  </component>
+</project>
\ No newline at end of file
diff --git a/setup.py b/setup.py
index 8aca7c5..17fb33f 100755
--- a/setup.py
+++ b/setup.py
@@ -25,7 +25,11 @@
     "py4j==0.10.7",
     "pyspark==2.3.1",
     "pytest==3.6.1",
-    "six==1.11.0"
+    "six==1.11.0",
+    "mlflow==0.4.2",
+    "graphviz==0.8.4",
+    "pyspark==2.1.0",
+    "numpy==1.14.3"
 ]
 setup(
     name="splicemachine",
diff --git a/splicemachine/ml/zeppelin.py b/splicemachine/ml/zeppelin.py
index e062825..23d8269 100644
--- a/splicemachine/ml/zeppelin.py
+++ b/splicemachine/ml/zeppelin.py
@@ -39,7 +39,8 @@ def handle_handlers(handler, *args, **kwargs):
     def log_metadata(self, handler, *args, **kwargs):
         if not self.run_uuid:
             with mlflow.start_run():
-                self.run_uuid = (mlflow.active_run().__dict__['_info'].__dict__['_run_uuid'])
+                self.run_uuid = (mlflow.active_run().__dict__[
+                                 '_info'].__dict__['_run_uuid'])
                 print("Logged using handler " + handler)
                 Run.handle_handlers(handler, *args, **kwargs)
         else:
@@ -77,7 +78,8 @@ def show_confusion_matrix(TP, TN, FP, FN):
     :param FN: False Negatives
     """
     confusion_rdd = sc.parallelize([['Predicted', TP, FN], ['Actual', FP, TN]])
-    confusion_matrix = sqlContext.createDataFrame(confusion_rdd, ['', 'Actual', 'Predicted'])
+    confusion_matrix = sqlContext.createDataFrame(
+        confusion_rdd, ['', 'Actual', 'Predicted'])
     confusion_matrix.show()
 
 
@@ -98,7 +100,8 @@ def experiment_maker(experiment_id):
                     e._experiment_id)  # use already created experiment
 
         if not found:
-            _id = mlflow.tracking.create_experiment(experiment_id)  # create new experiment
+            _id = mlflow.tracking.create_experiment(
+                experiment_id)  # create new experiment
             print('Success! Created Experiment')
             os.environ['MLFLOW_EXPERIMENT_ID'] = str(_id)  # use it
     else:
@@ -135,7 +138,7 @@ def input(self, predictions_dataframe):
                                                   self.prediction_column)  # Select the actual and the predicted labels
 
         self.avg_tp.append(pred_v_lab[(pred_v_lab.label == 1) & (
-                pred_v_lab.prediction == 1)].count())  # Add confusion stats
+            pred_v_lab.prediction == 1)].count())  # Add confusion stats
         self.avg_tn.append(
             pred_v_lab[(pred_v_lab.label == 0) & (pred_v_lab.prediction == 0)].count())
         self.avg_fp.append(
@@ -177,13 +180,14 @@ def get_results(self, output_type='dataframe'):
             return computed_metrics
 
         else:
-            metrics_row = Row('TPR', 'SPC', 'PPV', 'NPV', 'FPR', 'FDR', 'FNR', 'ACC', 'F1', 'MCC')
+            metrics_row = Row('TPR', 'SPC', 'PPV', 'NPV',
+                              'FPR', 'FDR', 'FNR', 'ACC', 'F1', 'MCC')
             computed_row = metrics_row(*computed_metrics.values())
             computed_df = sqlContext.createDataFrame([computed_row])
             return computed_df
 
 
-def print_horizantal_line(l):
+def print_horizontal_line(l):
     print("".join(['-' * l]))
 
 
@@ -235,8 +239,9 @@ def visualize(model, feature_column_names, label_names, tree_name, visual=True):
         time.sleep(3)
 
         print('You can find your visualization at "https://docs.google.com/gview?url=https'
-              '://<cluster_name>.splicemachine.io/assets/images/' + tree_name + '.pdf&embedded=tru'
-                                                                                'e#view=fith')
+              '://<cluster_name>.splicemachine.io/assets/images/' +
+              tree_name + '.pdf&embedded=tru'
+              'e#view=fith')
 
     @staticmethod
     def replacer(string, bad, good):
@@ -269,10 +274,13 @@ def add_node(dot, parent, node_hash, root, realroot=False):
                 dot.edge(node_hash, node_id)
             if root.get('children'):
                 if not root['children'][0].get('children'):
-                    DecisionTreeVisualizer.add_node(dot, root['name'], node_id, root['children'][0])
+                    DecisionTreeVisualizer.add_node(
+                        dot, root['name'], node_id, root['children'][0])
                 else:
-                    DecisionTreeVisualizer.add_node(dot, root['name'], node_id, root['children'][0])
-                    DecisionTreeVisualizer.add_node(dot, root['name'], node_id, root['children'][1])
+                    DecisionTreeVisualizer.add_node(
+                        dot, root['name'], node_id, root['children'][0])
+                    DecisionTreeVisualizer.add_node(
+                        dot, root['name'], node_id, root['children'][1])
 
     @staticmethod
     def parse(lines):
@@ -285,12 +293,16 @@ def parse(lines):
         while lines:
 
             if lines[0].startswith('If'):
-                bl = ' '.join(lines.pop(0).split()[1:]).replace('(', '').replace(')', '')
-                block.append({'name': bl, 'children': DecisionTreeVisualizer.parse(lines)})
+                bl = ' '.join(lines.pop(0).split()[1:]).replace(
+                    '(', '').replace(')', '')
+                block.append(
+                    {'name': bl, 'children': DecisionTreeVisualizer.parse(lines)})
 
                 if lines[0].startswith('Else'):
-                    be = ' '.join(lines.pop(0).split()[1:]).replace('(', '').replace(')', '')
-                    block.append({'name': be, 'children': DecisionTreeVisualizer.parse(lines)})
+                    be = ' '.join(lines.pop(0).split()[1:]).replace(
+                        '(', '').replace(')', '')
+                    block.append(
+                        {'name': be, 'children': DecisionTreeVisualizer.parse(lines)})
             elif not lines[0].startswith(('If', 'Else')):
                 block2 = lines.pop(0)
                 block.append({'name': block2})
@@ -314,5 +326,6 @@ def tree_json(tree):
                 break
             if not line:
                 break
-        res = [{'name': 'Root', 'children': DecisionTreeVisualizer.parse(data[1:])}]
+        res = [
+            {'name': 'Root', 'children': DecisionTreeVisualizer.parse(data[1:])}]
         return res[0]