Added learning curve for RMSE