rejection_sampling_stage() only returns z-values for AEP

[danmcgraw94]

Describe the bug
the function should return both z-value and AEPs. This isn't a difficult extra step, but it's annoying to deal with