The python file to build the dataset