diff --git a/recommend/utils/data_generator.py b/recommend/utils/data_generator.py
index e50b4e9..d3ef9cf 100644
--- a/recommend/utils/data_generator.py
+++ b/recommend/utils/data_generator.py
@@ -67,8 +67,6 @@
         sid = seed2idx[seed_id]
         user_items[uid].append(sid)
         user_times[uid].append(ts)
-    print(user_items)
-    print(user_times)
     with open(output_path, "w", encoding="utf-8") as f:
         for uid in sorted(user_items.keys()):
             items = " ".join(str(item) for item in user_items[uid])
@@ -76,12 +74,8 @@
             f.write(f"{uid}\t{items}\t{times}\n")
 
 
-def main():
+def build_user_seed_graph():
     download_rows, favorite_rows = fetch_data()
     records, user_set, seed_set = process_records(download_rows, favorite_rows)
     user2idx, seed2idx = build_id_maps(user_set, seed_set)
     group_and_write(records, user2idx, seed2idx)
-
-
-if __name__ == "__main__":
-    main()
\ No newline at end of file