added a few definitions to dataset.py