Add data preprocessing