updates to fix stats bugs, should be working with good RMSE now