diff --git a/example_feat_extract.py b/example_feat_extract.py
index a5bb904..2d49b0d 100644
--- a/example_feat_extract.py
+++ b/example_feat_extract.py
@@ -43,10 +43,14 @@ def feature_extraction_queue(feature_extractor, image_path, layer_names,
     :return:
     '''

+    #batch_size = 1
+
     # Add a list of images to process, note that the list is ordered.
-    image_files = utils.find_files(image_path, ("jpg", "png"))
-    num_images = min(len(image_files), num_images)
-    image_files = image_files[0:num_images]
+    image_files = utils.find_files(image_path, ("jpg", "jpeg", "png", "JPEG", "JPG"))
+    start_images = 29 * batch_size
+
+    num_images = batch_size * 1 #min(len(image_files), num_images)
+    image_files = image_files[start_images:start_images + num_images]