Adding mc pu reweighting

Adding some fixes that will make the code more convenient to use.