Prediction of thermal stratification in the cold leg of a PWR during a LOCA