diff --git a/brew/water_optimizer.m b/brew/water_optimizer.m
index f035817..fa39248 100644
--- a/brew/water_optimizer.m
+++ b/brew/water_optimizer.m
@@ -22,7 +22,7 @@
 ## Author: Jens <jens@orion>
 ## Created: 2023-06-08
 
-function water_optimizer (TEST_SUITE, k_ve)
+function water_optimizer (TEST_SUITE, k_ve, learning_rate)
 
 # Load constants
 water_constants()
@@ -76,13 +76,11 @@ T = create_mol([i_Ca i_Mg i_Na i_Cl i_S], [M_Ca_1_mmol M_Mg_1_mmol M_Na_1_mmol M
 if TEST_SUITE == 1,
 # Allowed substances, easy
 Pp = [M_EL_Ca; M_EL_Mg; M_EL_Na; M_EL_Cl; M_EL_S];
-learning_rate = 0.01;
 end
 
 if TEST_SUITE == 2,
 # Allowed substances, difficult
 Pp = [M_CaSO4; M_CaCl2; M_MgSO4; M_NaCl; M_HCl];
-learning_rate = 0.001;
 end
 
 # Init vecfit vars