add training and testing