diff --git a/examples/largedataset_cnn/process_data.py b/examples/largedataset_cnn/process_data.py
index 7d92d01..8489268 100644
--- a/examples/largedataset_cnn/process_data.py
+++ b/examples/largedataset_cnn/process_data.py
@@ -35,7 +35,6 @@
 
     return im
 
-
 def process_data(dataset_root, classes):
     # load class names
     with open(classes, 'r', encoding='utf-8') as f:
@@ -79,7 +78,7 @@
     return train_input_paths, train_labels, val_input_paths, val_labels
 
 def loaddata():
-    dataset_root = '/Dataset/Data/'
+    dataset_root = '/Dataset/Data/'  # need to set this path an argument
     classes = '/Dataset/classes.txt'
     return process_data(dataset_root, classes)