finalized one train stage