diff --git a/dataset/shapenet.py b/dataset/shapenet.py
index 27b9e96..7d079df 100644
--- a/dataset/shapenet.py
+++ b/dataset/shapenet.py
@@ -101,7 +101,7 @@ class CustomShapeNet(InMemoryDataset):
 
             for element in paths:
                 # This was build to filter all variations that aregreater then 25
-                pattern = re.compile('^((6[0-1]|[2-5][0-9])_\w+?\d+?|pc|\d+?_pc)\.(xyz|dat)$')
+                pattern = re.compile('^((6[0-1]|[1-5][0-9])_\w+?\d+?|pc|\d+?_pc)\.(xyz|dat)$')
                 if pattern.match(os.path.split(element)[-1]):
                     continue
                 else:
diff --git a/main.py b/main.py
index f2ad6dd..ad49a03 100644
--- a/main.py
+++ b/main.py
@@ -123,7 +123,8 @@ if __name__ == '__main__':
         print('Epoch {}, total epoches {}'.format(epoch+1, opt.nepoch))
 
         net.train()
-
+        # ToDo: We need different dataloader here to train the network in multiple iterations, maybe move the loop down
+        # for dataloader in ...
         for batch_idx, sample in enumerate(dataLoader):
             # points: (batch_size, n, 3)
             # labels: (batch_size, n)