- worked on using the training data