data collector class and training data added