add test, train