diff --git a/Dipolar-Gas-Simulator/+Scripts/run_locally.m b/Dipolar-Gas-Simulator/+Scripts/run_locally.m
index eb1778f..5ae87b6 100644
--- a/Dipolar-Gas-Simulator/+Scripts/run_locally.m
+++ b/Dipolar-Gas-Simulator/+Scripts/run_locally.m
@@ -552,12 +552,46 @@ Plotter.visualizeGSWavefunction(SaveDirectory, JobNumber)
 SaveDirectory = './Results/Data_3D/GradientDescent/aS_079_theta_000_phi_000_N_50000';
 JobNumber     = 0;
 Plotter.visualizeGSWavefunction(SaveDirectory, JobNumber)
+%%
+% Parameters you can set before the loop
+N         = 60000;
+theta     = 0;
+phi       = 0;
+JobNumber = 0;
+
+% Values of aS to loop over
+aS_values = [79.0 80.0 81.0 82.0 83.0 84.0 85.0 86.0 87.0 88.0 89.0 90.0];
+
+for aS = aS_values
+    % Format numbers
+    aS_str    = sprintf('%03.0f', aS);       
+    theta_str = sprintf('%03.0f', theta);    
+    phi_str   = sprintf('%03.0f', phi);      
+    N_str     = sprintf('%d', N);  
+
+    % Construct directory path
+    SaveDirectory = ['./Results/Data_3D/GradientDescent/aS_', aS_str, ...
+                     '_theta_', theta_str, '_phi_', phi_str, '_N_', N_str];
+
+    % Call the plotting function
+    try
+        Plotter.visualizeGSWavefunction(SaveDirectory, JobNumber)
+    catch ME
+        warning(ME.message)
+        continue;
+    end
+
+    % Pause to inspect plot before continuing
+    disp(['Plotted for aS = ', num2str(aS)])
+    pause(0.25)
+end
+
 %%
 SaveDirectory = './Results/Data_3D/GradientDescent/';
 JobNumber     = 1;
 Plotter.visualizeGSWavefunction(SaveDirectory, JobNumber)
 %% Visualize phase diagram
-load('phase_diagram_matrix.mat')
+load('phase_diagram_matrix_theta_0.mat')
 PhaseDiagramMatrix   = M;
 ScatteringLengths    = SCATTERING_LENGTH_RANGE;
 NumberOfAtoms        = NUM_ATOMS_LIST * 1E-5;