Stage flow improvements