R code for dataset