Add train.yml