diff --git a/data/imagenet10_dataloader.py b/data/imagenet10_dataloader.py
index b0a085e..21a208d 100644
--- a/data/imagenet10_dataloader.py
+++ b/data/imagenet10_dataloader.py
@@ -1,3 +1,4 @@
+#imports
 import torch
 import torchvision.datasets as datasets
 import torchvision.transforms as transforms
diff --git a/main.py b/main.py
index b9eba81..1088632 100644
--- a/main.py
+++ b/main.py
@@ -1,3 +1,4 @@
+#import
 import torch
 import config as cfg
 from imagenet10_dataloader import get_data_loaders