add train.hs