diff --git a/main.m b/main.m
index 018ece7..e80d490 100644
--- a/main.m
+++ b/main.m
@@ -60,7 +60,7 @@
 dt = 0.1;
 
 figure(2)
-CellsortICAplot('series', ica_filters, ica_sig, movm, tlims, dt, [], [], [1:20]);
+CellsortICAplot('series', ica_filters, ica_sig, movm, tlims, dt, [], [], [1:size(ica_segments,1)]);
 
 %% 4a. Segment contiguous regions within ICs
 %search for indepent segments, you have to adjust these values well:
@@ -94,7 +94,7 @@
 %% Show results
 %optionally, plot signals
 figure(2)
-CellsortICAplot('series', ica_filters, ica_sig, movm, tlims, dt, 1, 2, [1:20], spt, spc);
+CellsortICAplot('series', ica_filters, ica_sig, movm, tlims, dt, 1, 2, [1:size(ica_segments,1)], spt, spc);
 %% show signal of cells (cell_sig)
 %caluclate correlation matrix for cell_sig
 [ica_segments, cluster, subPlotCols, subPlotWidth, sortedIdx] = calcCorrEff(cell_sig, ica_segments);
@@ -121,4 +121,4 @@
 subPlotCols = size(cluster_new, 2);
 
 %redraw
-alternativePlot(ica_segments, cluster_new, cell_sig, subPlotCols, sortedIdx, data_r);
\ No newline at end of file
+alternativePlot(ica_segments, cluster_new, cell_sig, subPlotCols, sortedIdx, data_r);