optimized code #YOLO