added comments to dataset.py. about to start writing batching command in batching.py