diff --git a/projects/WDM/WDM_model.m b/projects/WDM/WDM_model.m
index 9563ef6..921c705 100644
--- a/projects/WDM/WDM_model.m
+++ b/projects/WDM/WDM_model.m
@@ -4,13 +4,12 @@
 run(fullfile(fileparts(mfilename('fullpath')),'WDM_settings.m'));
 
 num_realiz      = 50;
-% wavelengthplan  = calcWavelengthPlan(16,400e9,1310);
-wavelengthplan  = [1295,1305,1315,1325];
-link_length     = 2;
+wavelengthplan  = calcWavelengthPlan(16,400e9,1310);
+% wavelengthplan  = [1295,1305,1315,1325];
+link_length     = 10;
 pmd             = 0.1;
 gamma           = 0.0023;
 
-
 M               = 4;
 m = floor(log2(M)*10)/10;
 fsym            = 224e9;
@@ -115,14 +114,15 @@ for realiz = 1:num_realiz
     %%%%%% Fiber %%%%%%
     Opt_sig_wdm_fib=Opt_sig_wdm;
     
-    nSegments = 2;
+    segment_length = 1; % km
+    nSegments = link_length/segment_length;
     zdw = 1310;
     D_local = 0; %if ~=0, simulation uses "segmented fiber with d+,d-)
     randomize_D = true;
     Dvec = getDispersionVector(nSegments, D_local, zdw, randomize_D, random_key+realiz);
     for s = 1:nSegments
         
-        Opt_sig_wdm_fib = DP_Fiber("L",link_length/nSegments,"D",Dvec(s),"Dpmd",pmd,"Ds",0.07,...
+        Opt_sig_wdm_fib = DP_Fiber("L",segment_length,"D",Dvec(s),"Dpmd",pmd,"Ds",0.07,...
                         "beat_len",10,"corr_len",100,"dz",1,"manakov",0,...
                         "gamma",gamma,"lambda",zdw,"n_waveplates",10,"SS_dphimax",0.01,...
                         "SS_dzmax",50,"SS_dzmin",10,"X_alpha",0.3,"X_beta",0,"rng",1).process(Opt_sig_wdm_fib);