Cleanup code around classification MSE computation
Fix label on tree stats