1 files changed, 16 insertions, 1 deletions
diff --git a/src/main.c b/src/main.c
index f9cc292..4aaafac 100644
--- a/src/main.c
+++ b/src/main.c
@@ -126,7 +126,7 @@ bool_t flood(uint8_t* image, uint16_t* mask, size_t width, size_t height, size_t
   return FALSE;
 }
 
-int main()
+int main(int argc, char** argv)
 {
   //-----------------------------------------------
   //-LIST-FILES-IN-DIRECTORY-----------------------
@@ -144,6 +144,21 @@ int main()
     }
     free(file_list);
   }
+  if (argc > 1) {
+    file_list = lsdir(argv[1]);
+    if (file_list) {
+      size_t index = 0;
+      while (1) {
+	char* fname = file_list[index];
+	if (fname == NULL) {
+	  break;
+	}
+	printf("%s\n", fname);
+	free(file_list[index++]);
+      }
+      free(file_list);
+    }
+  }
   //-----------------------------------------------
 
   //-----------------------------------------------