Adding tests for classifying