Add training data gathering feature