diff --git a/data/imagenet10_dataloader.py b/data/imagenet10_dataloader.py
index b0a085e..6389658 100644
--- a/data/imagenet10_dataloader.py
+++ b/data/imagenet10_dataloader.py
@@ -3,6 +3,7 @@
 import torchvision.transforms as transforms
 import config
 
+# TODO: Add self-defined and flexible data generators to enable more diverse types of data augmentations.
 def get_data_loaders():
     print('==> Preparing Imagenet 10 class data..')
     # Define the paths to the training and validation datasets