Changed argument names for NewBatchGradBased