add train.cpp