Add initial version of the train class.