updated image sizes and minor code change