slight adjustment to data prep script