Adding batch size