Updated to have the image id as a parameter