Added some changes for the image sizing