Train a final model