updated data wrangling