fixed off by 1 error in batch size