add data preparation step