Changed batch size
Changed batch size