reverted back to easy computation of the R2
Improved visualization
improved description of this repo