diff --git a/functions/CellSortLoadTrials.m b/functions/CellSortLoadTrials.m
index 31d731c..c292008 100644
--- a/functions/CellSortLoadTrials.m
+++ b/functions/CellSortLoadTrials.m
@@ -51,9 +51,8 @@
 reference = mean(data_r_array, 3);
 qualityPlot = zeros(1,size(data_r_array, 3));
 parfor i = 1:size(data_r_array,3)
-    qualityPlot(i) = squeeze(mean(mean(data_r_array(:,:,i) - reference,1),2));
+    qualityPlot(i) = squeeze(mean(mean((data_r_array(:,:,i) - reference).^2,1),2));
 end
-qualityPlot = qualityPlot.^2;
 figure;
 plot(qualityPlot);
 for i=1:numberGreen
diff --git a/functions/CellSortLoadTrialsShiftCorrOnTrials.m b/functions/CellSortLoadTrialsShiftCorrOnTrials.m
index b548146..167442a 100644
--- a/functions/CellSortLoadTrialsShiftCorrOnTrials.m
+++ b/functions/CellSortLoadTrialsShiftCorrOnTrials.m
@@ -54,9 +54,8 @@
 reference = mean(data_r_array, 3);
 qualityPlot = zeros(1,size(data_r_array, 3));
 parfor i = 1:size(data_r_array,3)
-    qualityPlot(i) = squeeze(mean(mean(data_r_array(:,:,i) - reference,1),2));
+    qualityPlot(i) = squeeze(mean(mean((data_r_array(:,:,i) - reference).^2,1),2));
 end
-qualityPlot = qualityPlot.^2;
 figure;
 plot(qualityPlot);
 for i=1:numberGreen
diff --git a/main.m b/main.m
index a29b2b5..271c358 100644
--- a/main.m
+++ b/main.m
@@ -11,7 +11,7 @@
 CellSortXCorr(inFolder, outFolder, nSubIm, binningFactor);
 %% load files
 %go to corr folder
-outFolder = 'F:\Arbeit\testData\PCAICA';
+outFolder = 'C:\Users\ftvol\Desktop\data';
 %shifft correction on all frames
 %[data_g, data_r, frames, fn_green, fn_red, fileNamesGreen, fileNamesRed] = CellSortLoadTrials(outFolder);
 %shift correction only on trials