Minor fix

- Utilizing `chunk` parameter within `data.batch()`