Train model

Training the model.