Update example for latest version of YOLO