Changes for getting training data call.