add training data visualization