add train and test program