Added the 5th category to training set and re-ran the individual tests