Added model definition file and scripts to train.