added comments to dataset.py