Add in z reweighting