Adding mc pu reweighting

Adding some fixes to the code to make it more convenient.