Also trained with reweight ptZ