Fixed things
Merge pull request #10 from bendyorke/yolo

Fixed things