add batch size