Add train

Code needs train D: