Modified Yolo Class