Adding RMSE and other features