More coding, and less hard coding - I made a small change so memory isn't wasted in creating new file names (the YoloSwag files)