Added chat prediction for certain classes