Add all codes for training and test.