--- a/dmri.xml	Thu Nov 30 08:10:10 2017 -0500
+++ b/dmri.xml	Thu Nov 30 09:39:24 2017 -0500
@@ -5,6 +5,7 @@
     </requirements>
     <command detect_errors="exit_code"><![CDATA[
 python '$__tool_directory__/dmri.py'
+--drmi_dataset_type '$drmi_dataset_type_cond.drmi_dataset_type'
 --drmi_dataset $drmi_dataset
 --output_png '$output_png'
 --output_nifti1 '$output_nifti1'
@@ -12,6 +13,14 @@
 &>/dev/null
     ]]></command>
     <inputs>
+        <conditional name="drmi_dataset_type_cond">
+            <param name="drmi_dataset_type" type="select" label="Select download type">
+                <option value="dataset" selected="true">Dataset only</option>
+                <option value="dataset and label map">Dataset and label map</option>
+            </param>
+            <when option="dataset"/>
+            <when option="map"/>
+        </conditional>
         <param name="drmi_dataset" type="select" label="dRMI dataset">
             <option value="stanford_hardi" selected="true">stanford_hardi</option>
             <option value="sherbrooke_3shell">sherbrooke_3shell</option>
@@ -19,7 +28,7 @@
     </inputs>
     <outputs>
         <data name="output_png" format="png" label="${tool.name}: middle axial slice without (left) and with (right) diffusion weighting" />
-        <data name="output_nifti1" format="nifti1" label="${tool.name}: Nifti1 image" />
+        <data name="output_nifti1" format="nifti1" label="${tool.name}: ($drmi_dataset_type) Nifti1 image" />
     </outputs>
     <tests>
         <test>