Create train.h

The .h file that exposes the interface of train.cpp