diff --git a/App.py b/App.py
index 3e0da06..bedfcf6 100644
--- a/App.py
+++ b/App.py
@@ -21,7 +21,7 @@ if bfs3_flag or Astar or Astar2:
    Pole.stoneFlag = True
 TreeFlag=False
 nnFlag=True
-newModel=False
+newModel=True
 
 pygame.init()
 show_console=True
diff --git a/model.pth b/model.pth
deleted file mode 100644
index c177c79..0000000
Binary files a/model.pth and /dev/null differ
diff --git a/neuralnetwork.py b/neuralnetwork.py
index 709d320..61e79c6 100644
--- a/neuralnetwork.py
+++ b/neuralnetwork.py
@@ -9,7 +9,7 @@ from PIL import Image
 import random
 
 imageSize = (128, 128)
-labels = ['carrot', 'potato', 'tomato'] # musi być w kolejności alfabetycznej
+labels = ['carrot','corn', 'potato', 'tomato'] # musi być w kolejności alfabetycznej
 fertilizer = {labels[0]: 'kompost', labels[1]: 'saletra amonowa', labels[2]: 'superfosfat'}
 
 torch.manual_seed(42)