BUG Fix computation of RMSE
FIG Larger marker sizes
ENH Compute & display a confusion matrix