diff --git a/demos/octave/DetectingInteractionLags/sampleResults/sourceMemory-MIT.png b/demos/octave/DetectingInteractionLags/sampleResults/sourceMemory-MIT.png
new file mode 100755
index 0000000..3a7c8ed
Binary files /dev/null and b/demos/octave/DetectingInteractionLags/sampleResults/sourceMemory-MIT.png differ
diff --git a/demos/octave/DetectingInteractionLags/sampleResults/sourceMemory-TE.png b/demos/octave/DetectingInteractionLags/sampleResults/sourceMemory-TE.png
new file mode 100755
index 0000000..b0c2aac
Binary files /dev/null and b/demos/octave/DetectingInteractionLags/sampleResults/sourceMemory-TE.png differ
diff --git a/demos/octave/DetectingInteractionLags/transferWithSourceMemory.m b/demos/octave/DetectingInteractionLags/transferWithSourceMemory.m
index 423dd23..d5db571 100755
--- a/demos/octave/DetectingInteractionLags/transferWithSourceMemory.m
+++ b/demos/octave/DetectingInteractionLags/transferWithSourceMemory.m
@@ -147,15 +147,6 @@ function transferWithSourceMemory(savePlot)
 		hold off;
 		% Make figures ok for plotting:
 		legend('location', 'east');
-		if (savePlot)
-			xlabel('\eta', 'fontsize', 32);
-			ylabel('Information (bits)', 'fontsize', 32);
-			print('te.eps', '-deps', '-color');
-		else
-			% do these without fontsize to have more stable displays in octave
-			xlabel('\eta');
-			ylabel('Information (bits)');
-		end
 	else
 		% We're on Matlab - just make a quick plot
 		plot(deltas, teXnToYnplus1, 'rx', 'markersize', markersize); % Empirical: TE_{SPO}(X \rightarrow Y, 1);
@@ -169,6 +160,15 @@ function transferWithSourceMemory(savePlot)
 			'Empirical: TE_{SPO}(X \rightarrow y, 2)', 'Analytic: TE_{SPO}(X \rightarrow Y, 2)', ...
 			'Location', 'East');
 	end
+	if (savePlot)
+		xlabel('\eta', 'fontsize', 32);
+		ylabel('Information (bits)', 'fontsize', 32);
+		print('te.eps', '-depsc');
+	else
+		% do these without fontsize to have more stable displays in octave
+		xlabel('\eta');
+		ylabel('Information (bits)');
+	end
 
 	% Plot the MIT's
 	figure;
@@ -185,15 +185,6 @@ function transferWithSourceMemory(savePlot)
 		hold off;
 		% Make figures ok for plotting:
 		legend('location', 'east')
-		if (savePlot)
-			xlabel('\eta', 'fontsize', 32);
-			ylabel('Information (bits)', 'fontsize', 32);
-			print('mit.eps', '-deps', '-color');
-		else
-			% do these without fontsize to have more stable displays in octave
-			xlabel('\eta');
-			ylabel('Information (bits)');
-		end
 	else
 		% We're on Matlab - just make a quick plot
 		plot(deltas, mitXnToYnplus1, 'rx', 'markersize', markersize); % Empirical: MIT(X \rightarrow Y, 1)
@@ -207,6 +198,15 @@ function transferWithSourceMemory(savePlot)
 			'Empirical: MIT(X \rightarrow Y, 2)', 'Analytic: MIT(X \rightarrow Y, 2)', ...
 			'Location', 'East');
 	end
+	if (savePlot)
+		xlabel('\eta', 'fontsize', 32);
+		ylabel('Information (bits)', 'fontsize', 32);
+		print('mit.eps', '-depsc');
+	else
+		% do these without fontsize to have more stable displays in octave
+		xlabel('\eta');
+		ylabel('Information (bits)');
+	end
 
 	toc;
 end