diff --git a/Figures/base_infection_first_result.png b/Figures/base_infection_first_result.png
new file mode 100644
index 0000000000000000000000000000000000000000..93c1ff3730b7e23b97fedb72e682f85a1554099d
Binary files /dev/null and b/Figures/base_infection_first_result.png differ
diff --git a/SRI_matlab/main.m b/SRI_matlab/main.m
index fb9df2e2988780b5e14ee61519d76975008cda3d..ef695d13033cbc640269f5d76a2325404385055a 100644
--- a/SRI_matlab/main.m
+++ b/SRI_matlab/main.m
@@ -18,7 +18,7 @@ coeffs.QuarantineRate = 0;
 VaccinationVector = build_vacc_rate(0, 0, 0, N);
 % VaccinationVector = build_vacc_rate(0, 0, 50, N);
 
-m = model(S, I, Q, R, 1, N);
+m = model(S, I, Q, R, 0, N);
 
 t = 1:N;
 
diff --git a/SRI_matlab/step.m b/SRI_matlab/step.m
index c874baf51a46aa0b4b42337e4b2b3f37007d45b4..2353e56c81db450dcddd70cf04d69207fe323842 100644
--- a/SRI_matlab/step.m
+++ b/SRI_matlab/step.m
@@ -12,9 +12,9 @@ for i=6:length(X)
 end
 
 if length(X) > 4
-    infection = S * (1 - 25 * options.InfectionRate / N) ^ I;
-    % infecter = sum(X(5:length(X)) + I);
-    % infection = options.InfectionRate * S * infecter/N;
+    % infection = S * (1 - 25 * options.InfectionRate / N) ^ I;
+    infecter = sum(X(5:length(X)) + I);
+    infection = options.InfectionRate * S * infecter/N;
     new_waiting(1) = infection;
     dS = -infection;
     dI = X(length(X));