Changed error metric to RMSE