diff --git a/src/botsort.py b/src/botsort.py
index 9638f573..b429ce0c 100644
--- a/src/botsort.py
+++ b/src/botsort.py
@@ -5,7 +5,7 @@
 model = YOLO('yolov8n.pt')
 
 # Open the video file
-video_path = "test2_10seconds.mp4"
+video_path = "training_data.mp4"
 cap = cv2.VideoCapture(video_path)
 
 # Loop through the video frames
@@ -16,7 +16,7 @@
     if success:
         # Run YOLOv8 tracking on the frame, persisting tracks between frames
         # change tracker to botsort.yaml or bytetrack.yaml
-        results = model.track(frame, persist=True, tracker="bytetrack.yaml")
+        results = model.track(frame, persist=True, tracker="botsort.yaml")
 
         # Visualize the results on the frame
         annotated_frame = results[0].plot()
diff --git a/src/training_data.mp4 b/src/training_data.mp4
new file mode 100644
index 00000000..280d9d5b
Binary files /dev/null and b/src/training_data.mp4 differ
diff --git a/src/yolov8n.pt b/src/yolov8n.pt
new file mode 100644
index 00000000..d61ef50d
Binary files /dev/null and b/src/yolov8n.pt differ