Add test to check if the right amount of images are generated.
(Right now they are not)