Adding train model stuff