Added option to generate images of the rotated structure