diff --git a/code_buddy/main.py b/code_buddy/main.py
index 132da88..3bcf0e5 100644
--- a/code_buddy/main.py
+++ b/code_buddy/main.py
@@ -45,7 +45,9 @@ def get_all_files(repo):
 def is_image_file(filename):
     """Check if a file is an image based on its extension."""
     image_extensions = {'.jpg', '.jpeg', '.png', '.gif', '.bmp', '.tiff', '.webp', '.svg', '.ico', '.yaml', '.woff'}
-    return Path(filename).suffix.lower() in image_extensions
+    video_extensions = {'.webm', '.wmv', '.flv', '.ogv', '.mkv', '.avi', '.mov', '.mp4', '.f4v', '.vob', '.nsv', '.roq'}
+    extensions = image_extensions.union(video_extensions)
+    return Path(filename).suffix.lower() in extensions
 
 # is_file_name: takes a path to a file and returns False if file is excluded, file can be a relative path
 def is_excluded_file_name(file):
diff --git a/version.py b/version.py
index 0a12e44..0f1114b 100644
--- a/version.py
+++ b/version.py
@@ -1,2 +1,2 @@
 # version.py
-__version__ = "0.6.0"
+__version__ = "0.6.1"