Add train model and validation spec