add classify and training test