diff --git a/createNeuralNetworkDatabase.py b/createNeuralNetworkDatabase.py
new file mode 100644
index 0000000..4a36f49
--- /dev/null
+++ b/createNeuralNetworkDatabase.py
@@ -0,0 +1,37 @@
+import os
+import cv2
+import matplotlib
+import numpy as np
+import matplotlib.pyplot as plt
+
+path_potatoes = 'neural_network\\images\\potatoes'
+path_beetroot = 'neural_network\\images\\beetroot'
+size = 250
+
+#POTATOES
+training_data_potatoes = []
+for img in os.listdir(path_potatoes):
+    pic = cv2.imread(os.path.join(path_potatoes,img))
+    pic = cv2.cvtColor(pic,cv2.COLOR_BGR2RGB)
+    pic = cv2.resize(pic,(size,size))
+    training_data_potatoes.append([pic])
+
+np.save(os.path.join('neural_network','potatoes-dataset'),np.array(training_data_potatoes))
+
+saved_potatoes = np.load(os.path.join('neural_network','potatoes-dataset.npy'))
+
+#BEETROOT
+training_data_beetroot = []
+for img in os.listdir(path_beetroot):
+    pic = cv2.imread(os.path.join(path_beetroot,img))
+    pic = cv2.cvtColor(pic,cv2.COLOR_BGR2RGB)
+    pic = cv2.resize(pic,(size,size))
+    training_data_beetroot.append([pic])
+
+np.save(os.path.join('neural_network','beetroot-dataset'),np.array(training_data_beetroot))
+
+saved_potatoes = np.load(os.path.join('neural_network','beetroot-dataset.npy'))
+
+# for i in range(5):
+#     plt.imshow(np.array(training_data_potatoes[i]).reshape(size,size,3))
+#     plt.show()
\ No newline at end of file
diff --git a/main.py b/main.py
index 5125206..7c3fbd4 100644
--- a/main.py
+++ b/main.py
@@ -1,5 +1,3 @@
-import pygame
-
 import Board
 import FindPath
 import TractorAction
@@ -12,9 +10,12 @@ from TractorLoad import TillageUnit
 from constants import *
 from manualSteering import manualSteeringDriver
 import joblib
+import pygame
+
 
 pygame.init()
 
+
 display = pygame.display.set_mode((DISPLAY_SIZE_HORIZONTAL, DISPLAY_SIZE_VERTICAL))
 pygame.display.set_caption('Tractor')
 
diff --git a/treedata/DecisionTree.pkl b/treedata/DecisionTree.pkl
index c48082d..00fec2b 100644
Binary files a/treedata/DecisionTree.pkl and b/treedata/DecisionTree.pkl differ
diff --git a/treedata/mydecisiontree.png b/treedata/mydecisiontree.png
index 26aea1f..7fba3e5 100644
Binary files a/treedata/mydecisiontree.png and b/treedata/mydecisiontree.png differ