added call to training data for test