Adjusting the behaviour of ImageGenerator to be more customizable and better fit with the other generators.