diff --git a/Estimations/OpticalAccordionLattice.m b/Estimations/OpticalAccordionLattice.m
index 8d1a811..c49e0e8 100644
--- a/Estimations/OpticalAccordionLattice.m
+++ b/Estimations/OpticalAccordionLattice.m
@@ -118,7 +118,7 @@ ylabel('z [µm]', 'FontSize', 12);
 title(['I_{plane}(y, z) at x = 0, \theta = ' num2str(rad2deg(theta_val)) '^\circ'], 'FontSize', 14);
 set(gca, 'FontSize', 12, 'Box', 'on');
 
-% ================ Plot 1-D intensity profiles of lattice =================== %%
+%% ================ Plot 1-D intensity profiles of lattice =================== %%
 
 % Find indices closest to zero in y and z grids:
 [~, idx_y0] = min(abs(ygrid(1,:)));  % y=0 along columns
@@ -158,7 +158,7 @@ ylabel('Intensity');
 grid on;
 set(gca, 'FontSize', 12, 'Box', 'on');
 
-%% Plot scalar intensity to see effect from change in polarization
+%% ================ Plot scalar intensity to see effect from change in polarization =================== %%
 
 theta_val            = deg2rad(10); % half-angle between beams
 
@@ -224,8 +224,7 @@ title(['Contrast vs. Polarization Angle \gamma, \theta = ' num2str(rad2deg(theta
 grid on
 box on
 set(gca, 'FontSize', 14);
-%%
-% --- Contrast vs Theta for fixed gamma = 0 and gamma = pi/2 ---
+%% ================ Contrast vs Theta for fixed gamma = 0 and gamma = pi/2 =================== %%
 
 % Define range of theta values (in radians)
 theta_vals = linspace(deg2rad(1), deg2rad(45), 50);  % e.g., from 1° to 45°