Add train
Merge pull request #3 from TheUnnamedDude/train

Add train