change an error in the main.py
changes in the data_preprocessing