diff --git a/benchmarks/benchmark_performance.py b/benchmarks/benchmark_performance.py
index da616ab..9771844 100644
--- a/benchmarks/benchmark_performance.py
+++ b/benchmarks/benchmark_performance.py
@@ -35,9 +35,9 @@ def mlflow_context():
 @given(
     data=st.data(),
     feature_size=st.integers(min_value=1, max_value=10),
-    width=st.integers(min_value=MINIMUM_NETWORK_DIMENSION, max_value=20),
-    height=st.integers(min_value=MINIMUM_NETWORK_DIMENSION, max_value=20),
-    num_iterations=st.integers(min_value=10, max_value=50),
+    width=st.integers(min_value=MINIMUM_NETWORK_DIMENSION, max_value=100),
+    height=st.integers(min_value=MINIMUM_NETWORK_DIMENSION, max_value=100),
+    num_iterations=st.integers(min_value=10, max_value=100),
     initial_learning_rate=st.floats(min_value=1e-3, max_value=1.0, allow_nan=False, allow_infinity=False)
 )
 @settings(max_examples=10, deadline=None, suppress_health_check=(HealthCheck.too_slow,))
diff --git a/benchmarks/execution_time_comparison.png b/benchmarks/execution_time_comparison.png
new file mode 100644
index 0000000..b7858b1
Binary files /dev/null and b/benchmarks/execution_time_comparison.png differ
diff --git a/create_image.py b/create_image.py
new file mode 100644
index 0000000..0a36290
--- /dev/null
+++ b/create_image.py
@@ -0,0 +1,13 @@
+import numpy as np
+import matplotlib.pyplot as plt
+
+from models import train_kohonen_network, create_kohonen_params
+
+X = np.random.random((1000,3))
+
+IMAGE_SIZE = 10
+
+params = create_kohonen_params(X, IMAGE_SIZE, IMAGE_SIZE, 1000).unwrap()
+image_data = train_kohonen_network(X, params, use_mlflow=True).unwrap()
+
+plt.imsave(f'{IMAGE_SIZE}.png', image_data)
\ No newline at end of file