made yolo.py more readable