diff --git a/.idea/workspace.xml b/.idea/workspace.xml
index 19cbc3c..c31cabe 100644
--- a/.idea/workspace.xml
+++ b/.idea/workspace.xml
@@ -3,8 +3,8 @@
   <component name="ChangeListManager">
     <list default="true" id="56453584-72bd-49f4-a39c-fccf91ab20c6" name="Default Changelist" comment="">
       <change beforePath="$PROJECT_DIR$/.idea/workspace.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/workspace.xml" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/py.py" beforeDir="false" afterPath="$PROJECT_DIR$/py.py" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/resources/data.csv" beforeDir="false" afterPath="$PROJECT_DIR$/resources/data.csv" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/mydecisiontree.png" beforeDir="false" />
+      <change beforePath="$PROJECT_DIR$/treelearn.py" beforeDir="false" afterPath="$PROJECT_DIR$/treelearn.py" afterDir="false" />
     </list>
     <option name="SHOW_DIALOG" value="false" />
     <option name="HIGHLIGHT_CONFLICTS" value="true" />
@@ -131,6 +131,7 @@
       <workItem from="1621110649573" duration="32000" />
       <workItem from="1621110688412" duration="148000" />
       <workItem from="1621110850717" duration="377000" />
+      <workItem from="1621111315400" duration="72000" />
     </task>
     <servers />
   </component>
@@ -146,22 +147,22 @@
       <screen x="0" y="0" width="1920" height="1080" />
     </state>
     <state x="690" y="287" key="#com.intellij.refactoring.safeDelete.UnsafeUsagesDialog/0.0.1920.1080@0.0.1920.1080" timestamp="1617994393023" />
-    <state width="1879" height="295" key="GridCell.Tab.0.bottom" timestamp="1621111231927">
+    <state width="1879" height="295" key="GridCell.Tab.0.bottom" timestamp="1621111387013">
       <screen x="0" y="0" width="1920" height="1080" />
     </state>
-    <state width="1879" height="295" key="GridCell.Tab.0.bottom/0.0.1920.1080@0.0.1920.1080" timestamp="1621111231927" />
-    <state width="1879" height="295" key="GridCell.Tab.0.center" timestamp="1621111231927">
+    <state width="1879" height="295" key="GridCell.Tab.0.bottom/0.0.1920.1080@0.0.1920.1080" timestamp="1621111387013" />
+    <state width="1879" height="295" key="GridCell.Tab.0.center" timestamp="1621111387012">
       <screen x="0" y="0" width="1920" height="1080" />
     </state>
-    <state width="1879" height="295" key="GridCell.Tab.0.center/0.0.1920.1080@0.0.1920.1080" timestamp="1621111231927" />
-    <state width="1879" height="295" key="GridCell.Tab.0.left" timestamp="1621111231927">
+    <state width="1879" height="295" key="GridCell.Tab.0.center/0.0.1920.1080@0.0.1920.1080" timestamp="1621111387012" />
+    <state width="1879" height="295" key="GridCell.Tab.0.left" timestamp="1621111387012">
       <screen x="0" y="0" width="1920" height="1080" />
     </state>
-    <state width="1879" height="295" key="GridCell.Tab.0.left/0.0.1920.1080@0.0.1920.1080" timestamp="1621111231927" />
-    <state width="1879" height="295" key="GridCell.Tab.0.right" timestamp="1621111231927">
+    <state width="1879" height="295" key="GridCell.Tab.0.left/0.0.1920.1080@0.0.1920.1080" timestamp="1621111387012" />
+    <state width="1879" height="295" key="GridCell.Tab.0.right" timestamp="1621111387013">
       <screen x="0" y="0" width="1920" height="1080" />
     </state>
-    <state width="1879" height="295" key="GridCell.Tab.0.right/0.0.1920.1080@0.0.1920.1080" timestamp="1621111231927" />
+    <state width="1879" height="295" key="GridCell.Tab.0.right/0.0.1920.1080@0.0.1920.1080" timestamp="1621111387013" />
     <state width="1879" height="364" key="GridCell.Tab.1.bottom" timestamp="1619289752045">
       <screen x="0" y="0" width="1920" height="1080" />
     </state>
@@ -178,10 +179,10 @@
       <screen x="0" y="0" width="1920" height="1080" />
     </state>
     <state width="1879" height="364" key="GridCell.Tab.1.right/0.0.1920.1080@0.0.1920.1080" timestamp="1619289752045" />
-    <state x="0" y="0" key="com.intellij.ide.util.TipDialog" timestamp="1621110850962">
+    <state x="0" y="0" key="com.intellij.ide.util.TipDialog" timestamp="1621111315179">
       <screen x="0" y="0" width="1920" height="1080" />
     </state>
-    <state x="0" y="0" key="com.intellij.ide.util.TipDialog/0.0.1920.1080@0.0.1920.1080" timestamp="1621110850962" />
+    <state x="0" y="0" key="com.intellij.ide.util.TipDialog/0.0.1920.1080@0.0.1920.1080" timestamp="1621111315179" />
   </component>
   <component name="XDebuggerManager">
     <breakpoint-manager>
@@ -195,6 +196,6 @@
     </breakpoint-manager>
   </component>
   <component name="com.intellij.coverage.CoverageDataManagerImpl">
-    <SUITE FILE_PATH="coverage/SmartTractor$py.coverage" NAME="py Coverage Results" MODIFIED="1621111175660" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="true" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$" />
+    <SUITE FILE_PATH="coverage/SmartTractor$py.coverage" NAME="py Coverage Results" MODIFIED="1621111375675" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="true" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$" />
   </component>
 </project>
\ No newline at end of file
diff --git a/__pycache__/treelearn.cpython-37.pyc b/__pycache__/treelearn.cpython-37.pyc
index deabdf5..4f93d7d 100644
Binary files a/__pycache__/treelearn.cpython-37.pyc and b/__pycache__/treelearn.cpython-37.pyc differ
diff --git a/mydecisiontree.png b/mydecisiontree.png
deleted file mode 100644
index 4e0db11..0000000
Binary files a/mydecisiontree.png and /dev/null differ
diff --git a/treelearn.py b/treelearn.py
index fc0104e..201278d 100644
--- a/treelearn.py
+++ b/treelearn.py
@@ -15,6 +15,6 @@ def treelearn():
     data = tree.export_graphviz(dtree, out_file=None, feature_names=features)
     graph = pydotplus.graph_from_dot_data(data)
     graph.write_png(os.path.join('resources', 'mydecisiontree.png'))
-    img = pltimg.imread('mydecisiontree.png')
+    img = pltimg.imread(os.path.join('resources', 'mydecisiontree.png'))
     imgplot = plt.imshow(img)
     plt.show()
\ No newline at end of file