Set batch size to 10