Add updated R code