final version of the code with comments and shit, YOLO