Add test for classify