Add support for image dimensions