updated batch size