changes in the data sets now with fixed main.py