Add get_image_function