Add average rmse function