Adds rmse computation