add correct PU distribution
add correct PU distribution