Add more tests for classify() method