diff --git a/.gitignore b/.gitignore
index 6d6f29d..a18aaf7 100644
--- a/.gitignore
+++ b/.gitignore
@@ -217,4 +217,5 @@ __marimo__/
 
 config.toml
 browser_profile
-old/
\ No newline at end of file
+old/
+data/
\ No newline at end of file
diff --git a/workflow.py b/workflow.py
index 92a9f78..e211a6f 100644
--- a/workflow.py
+++ b/workflow.py
@@ -30,7 +30,7 @@ def format_file_tree(file_tree: list) -> str:
     return "\n".join(lines)
 
 
-def process_imdb_workflow(imdb_id: str, dl_dir: str = "/data/qb", jellyfin_dir: str = "/data/jellyfin"):
+def process_imdb_workflow(imdb_id: str, dl_dir: str = "/data/qb", jellyfin_base_dir: str = "/data/Jellyfin"):
     """
     Workflow to automatically find, download, and map torrents for an IMDb ID into a Jellyfin library.
     """
@@ -45,6 +45,12 @@ def process_imdb_workflow(imdb_id: str, dl_dir: str = "/data/qb", jellyfin_dir:
     title_dir = f"{title} ({year})"
     print(f"Found Title: {title_dir}")
 
+    movietype = imdb_info['data'].get('movietype', 'Movie')
+    if movietype == 'TV Series':
+        jellyfin_dir = f"{jellyfin_base_dir}/TV"
+    else:
+        jellyfin_dir = f"{jellyfin_base_dir}/Movie"
+
     print(f"\n=== [1] Searching Torrents for {imdb_id} ===")
     imdb_url = f"https://www.imdb.com/title/{imdb_id}/"
     search_res = search_mteam_torrents(imdb_url)
@@ -134,8 +140,13 @@ def process_imdb_workflow(imdb_id: str, dl_dir: str = "/data/qb", jellyfin_dir:
         print(f"Finished processing torrent: {tid}")
 
 if __name__ == "__main__":
-    import sys
+    import argparse
     
-    # Allow executing directly with `uv run workflow.py tt7742120`
-    test_id = sys.argv[1] if len(sys.argv) > 1 else "tt38872297"
-    process_imdb_workflow(test_id)
+    parser = argparse.ArgumentParser(description="Workflow to automatically find, download, and map torrents for an IMDb ID into Jellyfin.")
+    parser.add_argument("imdb_id", type=str, help="The IMDb ID to process (e.g., tt38872297)")
+    parser.add_argument("--dl-dir", type=str, default="/data/qb", help="The qBittorrent download directory")
+    parser.add_argument("--jellyfin-dir", type=str, default="/data/Jellyfin", help="The base Jellyfin library directory")
+    
+    args = parser.parse_args()
+    
+    process_imdb_workflow(args.imdb_id, dl_dir=args.dl_dir, jellyfin_base_dir=args.jellyfin_dir)