diff --git a/generate_dataset/simulator_ivc.py b/generate_dataset/simulator_ivc.py
index 99641cc..85819ce 100644
--- a/generate_dataset/simulator_ivc.py
+++ b/generate_dataset/simulator_ivc.py
@@ -22,7 +22,7 @@ def get_ivc(self, circuit: Circuit):
         if ssr == 2000000:
             ssr = 10000000
 
-        ssr*=2
+        ssr *= 2  # Increase quality
 
         circuit.R('cs', 'input', 'input_dummy', self.measurement_settings['internal_resistance'])
         circuit.AcLine('Current', circuit.gnd, 'input_dummy', rms_voltage=rms_voltage,
diff --git a/requirements.txt b/requirements.txt
index 6cd78d2..3db2ca6 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,3 +1,4 @@
 matplotlib
 PySpice
+numpy
 git+https://github.com/EPC-MSU/EPCore
\ No newline at end of file
diff --git a/test.py b/test.py
index b1d4e7a..2d09c0d 100644
--- a/test.py
+++ b/test.py
@@ -1,10 +1,16 @@
-# Cut screenshots from EPLab
-# import numpy as np
-# import os
-# import glob
+# Temp file for work with dataset
+# Will be removed soon
+
+import os
+import glob
+from epcore.filemanager import load_board_from_ufiv
+import numpy as np
+import matplotlib.pyplot as plt
+# from epcore.elements import Board
+# from epcore.filemanager import save_board_to_ufiv
+# from matplotlib.ticker import MultipleLocator
 # import cv2
-#
-#
+
 # files = glob.glob('C:\\dev\\ivc-circuit-detector\\dataset_human_compare\\original\\*\\*.png')
 #
 # for file in files:
@@ -14,15 +20,7 @@
 #     if not os.path.exists(os.path.split(filename)[0]):
 #         os.makedirs(os.path.split(filename)[0])
 #     cv2.imwrite(filename, img)
-import glob
 
-from epcore.elements import Board
-from epcore.filemanager import save_board_to_ufiv, load_board_from_ufiv
-import numpy as np
-import matplotlib.pyplot as plt
-from matplotlib.ticker import MultipleLocator
-import glob
-import os
 
 PATH = "dataset/measurement_1kHz_middle_5v/*.uzf"
 files = glob.glob(PATH)