--- a/dmri.py	Thu Nov 30 10:47:25 2017 -0500
+++ b/dmri.py	Thu Nov 30 10:51:01 2017 -0500
@@ -34,8 +34,8 @@
         os.rmdir(source_directory)
 
 # Get input data.
-input_dir = args.drmi_dataset
-if input_dir == 'stanford_hardi':
+input_dir = 'stanford_hardi'
+if args.drmi_dataset == 'stanford_hardi':
     if args.drmi_dataset_type == "dataset":
         fetch_stanford_hardi()
         fdwi = os.path.join(input_dir, 'HARDI150.nii.gz')
@@ -48,7 +48,7 @@
         fbval = os.path.join(input_dir, 'HARDI150.bval')
         fbvec = os.path.join(input_dir, 'HARDI150.bvec')
 else:
-    # input_dir == 'stanford_t1
+    # args.drmi_dataset == 'stanford_t1
         fetch_stanford_t1()
         fdwi = os.path.join(input_dir, 't1.nii.gz')
         fbval = os.path.join(input_dir, 't1.bval')