updates to datasets.py