Added ImageGenerator class