Small changes to R2D2 - a bit optimization and comments.