Rename test.py to dataset.py