added data wrangling stuff