Working version of classifying the test data