Update the container to fit the image inside