Came back to this image sorting project this afternoon and finally got it to work! Need to just tidy it up and remove the comments that over-explain (ChatGPT is very comment heavy at times, which is good!)