diff --git a/.idea/workspace.xml b/.idea/workspace.xml
index b7b3821..0188e16 100644
--- a/.idea/workspace.xml
+++ b/.idea/workspace.xml
@@ -2,22 +2,12 @@
 <project version="4">
   <component name="ChangeListManager">
     <list default="true" id="d2565aa5-b0c7-4b53-9b23-84bd42c9e5d7" name="Changes" comment="">
-      <change afterPath="$PROJECT_DIR$/.github/workflows/pytest_opt_cdrec.yml" afterDir="false" />
-      <change afterPath="$PROJECT_DIR$/.github/workflows/pytest_opt_iim.yml" afterDir="false" />
-      <change afterPath="$PROJECT_DIR$/.github/workflows/pytest_opt_mrnn.yml" afterDir="false" />
-      <change afterPath="$PROJECT_DIR$/.github/workflows/pytest_opt_stmvl.yml" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/.github/workflows/pytest_contamination.yml" beforeDir="false" afterPath="$PROJECT_DIR$/.github/workflows/pytest_contamination.yml" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/.github/workflows/pytest_imp_cdrec.yml" beforeDir="false" afterPath="$PROJECT_DIR$/.github/workflows/pytest_imp_cdrec.yml" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/.github/workflows/pytest_imp_iim.yml" beforeDir="false" afterPath="$PROJECT_DIR$/.github/workflows/pytest_imp_iim.yml" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/.github/workflows/pytest_imp_mrnn.yml" beforeDir="false" afterPath="$PROJECT_DIR$/.github/workflows/pytest_imp_mrnn.yml" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/.github/workflows/pytest_imp_stmvl.yml" beforeDir="false" afterPath="$PROJECT_DIR$/.github/workflows/pytest_imp_stmvl.yml" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/.github/workflows/pytest_loading.yml" beforeDir="false" afterPath="$PROJECT_DIR$/.github/workflows/pytest_loading.yml" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/.github/workflows/pytest_shap_explainer.yml" beforeDir="false" afterPath="$PROJECT_DIR$/.github/workflows/pytest_shap_explainer.yml" afterDir="false" />
       <change beforePath="$PROJECT_DIR$/.idea/workspace.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/workspace.xml" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/README.md" beforeDir="false" afterPath="$PROJECT_DIR$/README.md" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/assets/imputegab_logo.png" beforeDir="false" afterPath="$PROJECT_DIR$/assets/logo_imputegab.png" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/imputegap/contamination/README.md" beforeDir="false" afterPath="$PROJECT_DIR$/imputegap/contamination/README.md" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/tests/test_opti_bayesian_cdrec.py" beforeDir="false" afterPath="$PROJECT_DIR$/tests/test_opti_bayesian_cdrec.py" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/env/default_values.toml" beforeDir="false" afterPath="$PROJECT_DIR$/env/default_values.toml" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/imputegap/imputation/imputation.py" beforeDir="false" afterPath="$PROJECT_DIR$/imputegap/imputation/imputation.py" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/imputegap/optimization/algorithm_parameters.py" beforeDir="false" afterPath="$PROJECT_DIR$/imputegap/optimization/algorithm_parameters.py" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/requirements.txt" beforeDir="false" afterPath="$PROJECT_DIR$/requirements.txt" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/tests/test_opti_bayesian_iim.py" beforeDir="false" afterPath="$PROJECT_DIR$/tests/test_opti_bayesian_iim.py" afterDir="false" />
     </list>
     <option name="SHOW_DIALOG" value="false" />
     <option name="HIGHLIGHT_CONFLICTS" value="true" />
@@ -79,7 +69,7 @@
       <recent name="C:\Users\nquen\switchdrive\MST_MasterThesis\imputegap\assets" />
     </key>
   </component>
-  <component name="RunManager" selected="Python tests.Python tests in test_opti_bayesian_cdrec.py">
+  <component name="RunManager" selected="Python tests.Python tests in test_opti_bayesian_iim.py">
     <configuration name="runner_imputation" type="PythonConfigurationType" factoryName="Python" temporary="true" nameIsGenerated="true">
       <module name="imputegap" />
       <option name="INTERPRETER_OPTIONS" value="" />
@@ -154,7 +144,7 @@
       <option name="_new_targetType" value="&quot;PATH&quot;" />
       <method v="2" />
     </configuration>
-    <configuration name="Python tests in tests" type="tests" factoryName="Autodetect" temporary="true" nameIsGenerated="true">
+    <configuration name="Python tests in test_opti_bayesian_iim.py" type="tests" factoryName="Autodetect" temporary="true" nameIsGenerated="true">
       <module name="imputegap" />
       <option name="INTERPRETER_OPTIONS" value="" />
       <option name="PARENT_ENVS" value="true" />
@@ -165,7 +155,7 @@
       <option name="ADD_SOURCE_ROOTS" value="true" />
       <EXTENSION ID="PythonCoverageRunConfigurationExtension" runner="coverage.py" />
       <option name="_new_additionalArguments" value="&quot;&quot;" />
-      <option name="_new_target" value="&quot;$PROJECT_DIR$/tests&quot;" />
+      <option name="_new_target" value="&quot;$PROJECT_DIR$/tests/test_opti_bayesian_iim.py&quot;" />
       <option name="_new_targetType" value="&quot;PATH&quot;" />
       <method v="2" />
     </configuration>
@@ -173,16 +163,16 @@
       <item itemvalue="Python.runner_imputation" />
       <item itemvalue="Python.runner_optimization" />
       <item itemvalue="Python tests.Python tests in test_opti_bayesian_cdrec.py" />
+      <item itemvalue="Python tests.Python tests in test_opti_bayesian_iim.py" />
       <item itemvalue="Python tests.Python tests in test_explainer.py" />
-      <item itemvalue="Python tests.Python tests in tests" />
     </list>
     <recent_temporary>
       <list>
+        <item itemvalue="Python tests.Python tests in test_opti_bayesian_iim.py" />
         <item itemvalue="Python tests.Python tests in test_opti_bayesian_cdrec.py" />
         <item itemvalue="Python.runner_imputation" />
         <item itemvalue="Python.runner_optimization" />
         <item itemvalue="Python tests.Python tests in test_explainer.py" />
-        <item itemvalue="Python tests.Python tests in tests" />
       </list>
     </recent_temporary>
   </component>
@@ -211,7 +201,7 @@
       <workItem from="1727357029326" duration="44000" />
       <workItem from="1727358317602" duration="1867000" />
       <workItem from="1727361217072" duration="594000" />
-      <workItem from="1727434411640" duration="8444000" />
+      <workItem from="1727434411640" duration="8803000" />
     </task>
     <servers />
   </component>
@@ -235,7 +225,7 @@
     <SUITE FILE_PATH="coverage/imputegap$test.coverage" NAME="test Coverage Results" MODIFIED="1727099350719" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="true" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$" />
     <SUITE FILE_PATH="coverage/imputegap$test__1_.coverage" NAME="test (1) Coverage Results" MODIFIED="1727099954120" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="true" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$/imputegap" />
     <SUITE FILE_PATH="coverage/imputegap$_loading.coverage" NAME="_loading Coverage Results" MODIFIED="1726134442146" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="true" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$/imputegap/loading" />
-    <SUITE FILE_PATH="coverage/imputegap$.coverage" NAME=" Coverage Results" MODIFIED="1727443305357" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="true" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$/tests" />
+    <SUITE FILE_PATH="coverage/imputegap$.coverage" NAME=" Coverage Results" MODIFIED="1727444189590" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="true" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$/tests" />
     <SUITE FILE_PATH="coverage/imputegap$runner_optimization.coverage" NAME="runner_optimization Coverage Results" MODIFIED="1727437737184" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="true" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$/imputegap" />
     <SUITE FILE_PATH="coverage/imputegap$runner_imputation.coverage" NAME="runner_imputation Coverage Results" MODIFIED="1727439213786" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="true" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$/imputegap" />
   </component>
diff --git a/env/default_values.toml b/env/default_values.toml
index 618eb45..877848a 100644
--- a/env/default_values.toml
+++ b/env/default_values.toml
@@ -10,7 +10,7 @@ gamma = 0.85
 alpha = 7
 
 [iim]
-neighbor = 10
+learning_neighbors = 10
 algorithm_code = "iim 2"
 
 [mrnn]
diff --git a/imputegap/imputation/__pycache__/imputation.cpython-312.pyc b/imputegap/imputation/__pycache__/imputation.cpython-312.pyc
index 9bc19cb..734397f 100644
Binary files a/imputegap/imputation/__pycache__/imputation.cpython-312.pyc and b/imputegap/imputation/__pycache__/imputation.cpython-312.pyc differ
diff --git a/imputegap/imputation/imputation.py b/imputegap/imputation/imputation.py
index 79603de..727ddb8 100644
--- a/imputegap/imputation/imputation.py
+++ b/imputegap/imputation/imputation.py
@@ -32,7 +32,7 @@ def load_parameters(query="default", algorithm="cdrec"):
         if not os.path.exists(filepath):
             filepath = filepath[:1]
 
-        with open(filepath, "r") as file:
+        with open(filepath, "r") as _:
             config = toml.load(filepath)
 
         params = None
@@ -47,9 +47,9 @@ def load_parameters(query="default", algorithm="cdrec"):
             alpha = int(config['stmvl']['alpha'])
             params = (window_size, gamma, alpha)
         elif algorithm == "iim":
-            neighbors = int(config['iim']['neighbor'])
+            learning_neighbors = int(config['iim']['learning_neighbors'])
             algo_code = config['iim']['algorithm_code']
-            params = (neighbors, algo_code)
+            params = (learning_neighbors, algo_code)
         elif algorithm == "mrnn":
             hidden_dim = int(config['mrnn']['hidden_dim'])
             learning_rate = float(config['mrnn']['learning_rate'])
diff --git a/imputegap/optimization/__pycache__/algorithm_parameters.cpython-312.pyc b/imputegap/optimization/__pycache__/algorithm_parameters.cpython-312.pyc
index 2d874f2..72f742a 100644
Binary files a/imputegap/optimization/__pycache__/algorithm_parameters.cpython-312.pyc and b/imputegap/optimization/__pycache__/algorithm_parameters.cpython-312.pyc differ
diff --git a/imputegap/optimization/algorithm_parameters.py b/imputegap/optimization/algorithm_parameters.py
index 7fb8cf1..f186373 100644
--- a/imputegap/optimization/algorithm_parameters.py
+++ b/imputegap/optimization/algorithm_parameters.py
@@ -42,7 +42,7 @@
 # Define the parameter names for each algorithm
 PARAM_NAMES = {
     'cdrec': ['rank', 'epsilon', 'iteration'],
-    'iim': ['neighbor'],
+    'iim': ['learning_neighbors'],
     'mrnn': ['hidden_dim', 'learning_rate', 'iterations', 'seq_len' ],
     'stmvl': ['window_size', 'gamma', 'alpha']
 }
diff --git a/requirements.txt b/requirements.txt
index a873ff3..1c9e028 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,4 @@
-numpy==1.13.3
+numpy==1.21.5
 pandas==2.0.3
 matplotlib==3.7.5
 toml==0.10.2
diff --git a/tests/__pycache__/test_opti_bayesian_iim.cpython-312.pyc b/tests/__pycache__/test_opti_bayesian_iim.cpython-312.pyc
index efdcba2..f189be8 100644
Binary files a/tests/__pycache__/test_opti_bayesian_iim.cpython-312.pyc and b/tests/__pycache__/test_opti_bayesian_iim.cpython-312.pyc differ
diff --git a/tests/test_opti_bayesian_iim.py b/tests/test_opti_bayesian_iim.py
index 69639c0..2d49dcd 100644
--- a/tests/test_opti_bayesian_iim.py
+++ b/tests/test_opti_bayesian_iim.py
@@ -55,8 +55,7 @@ def test_optimization_bayesian_stmvl(self):
         params = Imputation.load_parameters(query="default", algorithm=algorithm)
         params_optimal = (optimal_params['neighbor'], "iim 2")
 
-        _, metrics_optimal = Imputation.Regression.iim_imputation(ground_truth=gap.ts, contamination=ts_contaminated,
-                                                 params=params_optimal)
+        _, metrics_optimal = Imputation.Regression.iim_imputation(ground_truth=gap.ts, contamination=ts_contaminated, params=params_optimal)
         _, metrics_default = Imputation.Regression.iim_imputation(ground_truth=gap.ts, contamination=ts_contaminated, params=params)
 
         Optimization.save_optimization(optimal_params=optimal_params, algorithm=algorithm+"_test")