Add `image_name` argument