diff --git a/src/curve.m b/src/curve.m
index f58e244de7d34f5677558d5d09b4b8859865ba38..69c90ccbceb846dc5afd8aa4359e02c4ad84dce8 100644
--- a/src/curve.m
+++ b/src/curve.m
@@ -13,6 +13,7 @@ b=[
    1
    2.5
 ];
+func = 'S'; % steepFunctionNonsmooth
 
 x = -.1:.005:.4;
 y = .9:.005:1.1;
@@ -23,7 +24,7 @@ ylabel('y')
 tic
 for i=1:length(y)
   in = [ X(i,:); Y(i,:) ];
-  f(i,:) = duneevaluate(A,b, 'S', in);
+  f(i,:) = duneevaluate(A,b,func, in);
 end
 clear X Y;
 toc
@@ -40,11 +41,11 @@ vecs(2,:) = [0; 1]; % Start
 
 for j = 2:steps+1 % Something random
     % Zero is code for 'SteepFunctionNonsmooth'
-    vecs(j+1,:) = duneminimise(A,b, 'S', vecs(j,:));
+    vecs(j+1,:) = duneminimise(A,b,func,vecs(j,:));
     diffs(j,:) = vecs(j+1,:) - vecs(j,:);
     line([vecs(j,1) vecs(j+1,1)], ...
 	 [vecs(j,2) vecs(j+1,2)], ...
-       [duneevaluate(A,b, 'S', vecs(j,:)') duneevaluate(A,b, 'S', vecs(j+1,:)')], ...
+       [duneevaluate(A,b,func,vecs(j,:)') duneevaluate(A,b,func,vecs(j+1,:)')], ...
        'color', 'y');
 end
 
diff --git a/src/zigzag.m b/src/zigzag.m
index 121210074656cf623a5c30b8c0583f2cf4bfc980..13f2b54ca179154e6470803b0f38ee29431c1b9b 100644
--- a/src/zigzag.m
+++ b/src/zigzag.m
@@ -13,6 +13,7 @@ b=[
    1
    2
 ];
+func = 's'; % sampleFunction
 
 x = -50:1:300;
 y = -125:1:50;
@@ -23,7 +24,7 @@ ylabel('y')
 tic
 for i=1:length(y)
   in = [ X(i,:); Y(i,:) ];
-  f(i,:) = duneevaluate(A,b, 's', in);
+  f(i,:) = duneevaluate(A,b,func,in);
 end
 clear X Y;
 toc
@@ -40,11 +41,11 @@ vecs(2,:) = [279; 0]; % Start
 
 for j = 2:steps+1 % Something random
     % One is code for 'SampleFunction'
-    vecs(j+1,:) = duneminimise(A,b, 's', vecs(j,:));
+    vecs(j+1,:) = duneminimise(A,b,func,vecs(j,:));
     diffs(j,:) = vecs(j+1,:) - vecs(j,:);
     line([vecs(j,1) vecs(j+1,1)], ...
 	 [vecs(j,2) vecs(j+1,2)], ...
-       [duneevaluate(A,b, 's', vecs(j,:)') duneevaluate(A,b, 's', vecs(j+1,:)')], ...
+       [duneevaluate(A,b,func,vecs(j,:)') duneevaluate(A,b,func,vecs(j+1,:)')], ...
        'color', 'y');
 end