Changed how the image size argument works