add class prediction test