Update preprocess.py