Added temp solution for image sizes