--- a/dmri.py	Fri Nov 03 13:51:40 2017 -0400
+++ b/dmri.py	Fri Nov 03 13:55:26 2017 -0400
@@ -32,5 +32,5 @@
 pyplot.imshow(data[:, :, axial_middle, 0].T, cmap='gray', origin='lower')
 pyplot.subplot(1, 2, 2).set_axis_off()
 pyplot.imshow(data[:, :, axial_middle, 10].T, cmap='gray', origin='lower')
-pyplot.show()
-pyplot.savefig(args.output, bbox_inches='tight')
+pyplot.savefig('data.png', bbox_inches='tight')
+shutil.move('data.png', args.output)