diff --git a/.gitignore b/.gitignore
index 6c881e8..784ac7a 100644
--- a/.gitignore
+++ b/.gitignore
@@ -7,4 +7,5 @@ docker-compose.yml~
 rb_ws/src/buggy/bags/*                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                   
 *.bag
 vision/data/*
-.vision/*
\ No newline at end of file
+.vision/*
+runs/*
\ No newline at end of file
diff --git a/vision/best.pt b/vision/best.pt
new file mode 100644
index 0000000..46e76ba
Binary files /dev/null and b/vision/best.pt differ
diff --git a/vision/ultralytics-train.py b/vision/ultralytics-train.py
index 9e67c6a..55bcb8d 100644
--- a/vision/ultralytics-train.py
+++ b/vision/ultralytics-train.py
@@ -2,7 +2,9 @@
 
 # Load a model
 model = YOLO("yolo11n.pt")  # load a pretrained model (recommended for training)
+yaml_path = "C:\\Users\\Mehul Goel\\Documents\\robobuggy-software\\vision\\data\\Buggy Vision.v1i.yolov11\\data.yaml"
+
 
 # Train the model with MPS
-results = model.train(data="/Users/mehulgoel/Documents/robobuggy-software/vision/data/Buggy Vision.v1i.yolov11/data.yaml", epochs=100, imgsz=640, device="mps")
+results = model.train(data=yaml_path, epochs=100, imgsz=640)
 print(results)
\ No newline at end of file