diff --git a/.idea/neuralNetwork.iml b/.idea/neuralNetwork.iml
index 74d515a..6690b72 100644
--- a/.idea/neuralNetwork.iml
+++ b/.idea/neuralNetwork.iml
@@ -7,4 +7,7 @@
     <orderEntry type="inheritedJdk" />
     <orderEntry type="sourceFolder" forTests="false" />
   </component>
+  <component name="PackageRequirementsSettings">
+    <option name="removeUnused" value="true" />
+  </component>
 </module>
\ No newline at end of file
diff --git a/requirements.txt b/requirements.txt
index 9c5cbc3..11c121c 100644
Binary files a/requirements.txt and b/requirements.txt differ
diff --git a/test_compare_tensorflow.py b/test_compare_tensorflow.py
index 3f1f03b..af10b42 100644
--- a/test_compare_tensorflow.py
+++ b/test_compare_tensorflow.py
@@ -20,7 +20,7 @@ def test_digit_tensorflow():
     network.add(Layer(neurons_count=10))
     network.compile(loss=SquareError,optimizer=Adam(0.0002),metric=Accuracy)
 
-    network.fit(X_train,y_train,epochs=3,batch_size=1, validation_split=0.1)
+    network.fit(X_train,y_train,epochs=3,batch_size=32, validation_split=0.1)
 
     model = tf.keras.models.Sequential([
         tf.keras.layers.Flatten(input_shape=(28, 28)),