Refactored and added training data sample.