small changes yolo