Fixed bugs with trfStats and classes