Last minute changes YOLO