Added parameter for an image generator