Add css class to handle image sizes properly