Add preprocess function