Initial commit for TrainPuzzle solution