diff --git a/Zadanie_06_training.py b/Zadanie_06_training.py
index 2867720..6a5352a 100644
--- a/Zadanie_06_training.py
+++ b/Zadanie_06_training.py
@@ -10,6 +10,8 @@ from tensorflow.keras.optimizers import Adam
 from sacred.observers import FileStorageObserver
 from sacred import Experiment
 from datetime import datetime
+import os
+import pymongo
 
 
 wine=pd.read_csv('train.csv')