Yolov5 Detecting Script improvements

I need two adjustments to a script that works.
- The class should be visible in the saved images
- the script should be able to handle several classes (currently only for 1 class)