added YOLO comments