added PU weight functionality