Add end train condition in SAE train.