add categories 5-8 to the test set