updated data prep script some more