Data wrangling and plotting in R