Use test_set to train