Let's abstract the wait out (keep it in this class) and give it a parameter of images to wait for.
Then we can call it from the test when we know exactly how many we need to wait for.
In this launch we'll just keep the wait for the progress_bar to clear. I think that will be a good compromise! What do you think?