Add minor changes for image size