Add script (in progress) to generate an image for each stage