test/train added