Add testing and training method.