diff --git a/PlotNeuronOutlines.m b/PlotNeuronOutlines.m
index e63f3e2..afa3557 100644
--- a/PlotNeuronOutlines.m
+++ b/PlotNeuronOutlines.m
@@ -1,9 +1,8 @@
-function [ output_args ] = PlotNeuronOutlines(PixelList,Xdim,Ydim,clusterlist)
-%UNTITLED Summary of this function goes here
-%   Detailed explanation goes here
-%figure;
+function [] = PlotNeuronOutlines(PixelList,Xdim,Ydim,clusterlist)
+%PlotNeuronOutlines(PixelList,Xdim,Ydim,clusterlist)
+figure;
 
-if (nargin < 4)
+if (~exist('clusterlist'))
     clusterlist = 1:length(PixelList)
 end
 
@@ -14,7 +13,7 @@
 colors = rand(length(clusterlist),3);
 
 for i = 1:length(clusterlist)
-    i/length(clusterlist)
+    
     temp = zeros(Xdim,Ydim);
     temp(PixelList{i}) = 1;
     b = bwboundaries(temp);
diff --git a/Tenaspis.m b/Tenaspis.m
index 0d47702..4d9c19d 100644
--- a/Tenaspis.m
+++ b/Tenaspis.m
@@ -23,7 +23,7 @@
 save Blobthresh.mat thresh;
 
 %% Step 4 (optional): Create the mask
-if (nargin < 2)
+if (~exist('mask'))
     EstimateBlobs('D1Movie.h5',0,thresh);
     beep;
     MakeBlobMask();