added testing and training data
started implementing the assignment