Cleaning the R code of augmentation a bit, adding some comments