Add reporter parameter
Add rmse computation
Add momentum