Added image generator to check result