added test train both