Update the testing & training data for assignment