Restored hardcoded image size
Added returned value of image generator