Add data preparation script