Add a reweighting step for the initial estimate