Create datasets by passing around *args.
Fix off-by-one error in splitting batches.