Stratification of magnesium sulphate dose