preprocess.py modified