Update main.cpp

I changed data set folder structures