Update to Data Preparation Script