diff --git a/matlab/+SSF/draw_ssf_ncd.m b/matlab/+SSF/draw_ssf_ncd.m
index 91e362ab3c6d359d186ff25439723c323b6a5b9d..b58fff8210aef06859c56e7448107592c8e79ed4 100644
--- a/matlab/+SSF/draw_ssf_ncd.m
+++ b/matlab/+SSF/draw_ssf_ncd.m
@@ -1,20 +1,12 @@
-% load('agne_72_8bpp.mat')
-ROOT = '/g/leverjs/Olivier/Agne/march_2023_20x';
-flist = dir(fullfile(ROOT,'*.LEVER'));
+function [csf_mean,csf_std] = draw_ssf_ncd(ROOT,classList,A,K)
 
-classList = get_ssClasses(flist);
+flist = dir(fullfile(ROOT,'*.LEVER'));
 [classUnique,ia,ic]= unique(classList);
 idxTrue = ic;
-% targetClass = {'AKT1_E17K','WT','PIK3CA_E545K'}
-% idx = find(cellfun(@(x) ~isempty(find(strcmp(targetClass,x))),className));
-% A=d(idx,idx);
-% className = className(idx);
 
-A = d;
-K=6;
 [idx,Y] = Cluster.SpectralCluster(A,K);
 
-[csf_mean, csf_std] = CSF.csf_spatial(Y,idxTrue)
+[csf_mean, csf_std] = CSF.csf_spatial(Y,idxTrue);
 
 % Y = tsne(Y,'Algorithm','exact','NumDimensions',3);
 clf;hold off;